Python Pandas - Filtering few rows from a DataFrame on the basis of sum

To filter few rows from DataFrame on the basis of sum, we have considered an example with Student Marks. We need to calculate the sum of a particular subject wherein the total is more than 200 i.e. the total of all 3 students in that particular subject is more than 200. In this way we can fiter our rows with total less than 200.

At first, let us create a DataFrame with 3 columns i.e. records of 3 students −

dataFrame = pd.DataFrame({'Jacob_Marks': [95, 90, 70, 85, 88],'Ted_Marks': [60, 50, 65, 85, 70],'Jamie_Marks': [77, 76, 60, 45, 50]})

Filtering on the basis of rows. Fetching rows with total greater than 200 for all the 3 students −

dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

Example

Following is the complete code −

import pandas as pd

# create a dataframe with 3 columns
dataFrame = pd.DataFrame({'Jacob_Marks': [95, 90, 70, 85, 88],'Ted_Marks': [60, 50, 65, 85, 70],'Jamie_Marks': [77, 76, 60, 45, 50]})

print"Dataframe...\n",dataFrame

# filtering on the basis of rows
# fetching rows with total greater than 200 for all the 3 students
dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]

# dataframe
print"Updated Dataframe...\n",dataFrame

Output

This will produce the following output −

Dataframe...
   Jacob_Marks  Jamie_Marks   Ted_Marks
0           95           77         60
1           90           76         50
2           70           60         65
3           85           45         85
4           88           50         70
Updated Dataframe...
   Jacob_Marks   Jamie_Marks   Ted_Marks
0           95            77         60
1           90            76         50
3           85            45         85
4           88            50         70

AmitDiwan

Published on 13-Sep-2021 08:33:28

• Related Questions & Answers
• Python Pandas - Filtering columns from a DataFrame on the basis of sum
• Python Pandas – How to select DataFrame rows on the basis of conditions
• Python - Sum only specific rows of a Pandas Dataframe
• Python Pandas - Select a subset of rows from a dataframe
• Python Pandas - Display specific number of rows from a DataFrame
• Select rows from a Pandas DataFrame based on column values
• Python - Summing all the rows of a Pandas Dataframe
• Python - Ranking Rows of Pandas DataFrame
• Python - How to drop the null rows from a Pandas DataFrame
• Python - Drop specific rows from multiindex Pandas Dataframe
• How to get few rows from a Series in Pandas?
• Python Pandas - How to select multiple rows from a DataFrame
• Python Pandas - How to select rows from a DataFrame by integer location
• Use a list of values to select rows from a Pandas DataFrame
• Delete the first three rows of a DataFrame in Pandas