Python Pandas - 根据总和从 DataFrame 中过滤几行
为了根据总和从DataFrame中过滤几行，我们考虑了一个带有学生标记的示例。我们需要计算特定科目的总和，其中总人数超过200，即该特定科目的所有3名学生的总人数超过200。通过这种方式，我们可以调整总人数少于200的行。
首先，让我们创建一个包含3列的DataFrame，即3个学生的记录-
dataFrame = pd.DataFrame({'Jacob_Marks': [95, 90, 70, 85, 88],'Ted_Marks': [60, 50, 65, 85, 70],'Jamie_Marks': [77, 76, 60, 45, 50]})基于行过滤。为所有3名学生获取总行数大于200的行-
dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]
示例
以下是完整的代码-
import pandas as pd
#createadataframewith3columns
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
#filteringonthebasisofrows
#fetchingrowswithtotalgreaterthan200forallthe3students
dataFrame = dataFrame[dataFrame.sum(axis=1) > 200]
#dataframe
print"Updated Dataframe...\n",dataFrame输出结果这将产生以下输出-
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