Python Pandas - 根据总和从 DataFrame 中过滤几行

pythonserver side programmingprogramming更新于 2025/10/28 3:24:17

为了根据总和从 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

# 创建一个包含 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]})

print"Dataframe...\n",dataFrame

# 根据行进行过滤
# 获取所有 3 名学生的总数大于 200 的行
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

相关文章


有用资源