Python Pandas - 根据总和过滤 DataFrame 中的列

pythonserver side programmingprogramming更新于 2025/10/29 16:12:17

要根据列的总和进行过滤,我们使用 loc() 方法。在这里,在我们的示例中,我们将每个学生的分数相加,以获得分数高于 400 即 80% 的学生列。

首先,创建一个包含学生记录的 DataFrame。我们有 3 名学生的成绩记录,即 3 列 −

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

根据列进行过滤。获取总分高于 400 的学生 −

dataFrame = dataFrame.loc[:, dataFrame.sum(axis=0) > 400]

示例

以下是完整代码 −

import pandas as pd

# 创建一个包含 3 列的数据框
dataFrame = pd.DataFrame({
    'Jacob_Marks': [95, 90, 75, 85, 88],'Ted_Marks': [60, 50, 65, 85, 70],'Jamie_Marks': [77, 76, 65, 45, 50]})

print"Dataframe...\n",dataFrame

# 根据列进行过滤
# 获取总分高于 400 的学生
dataFrame = dataFrame.loc[:, dataFrame.sum(axis=0) > 400]

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

输出

这将产生以下输出 −

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

相关文章


有用资源