如何遍历dataframe中的一列,并按照特定的分类方式,对dataframe中的另一列求和?
时间: 2023-06-04 07:04:53 浏览: 217
pandas对dataFrame中某一个列的数据进行处理的方法
你可以使用 Pandas 库中的 groupby() 函数实现此目的。首先,使用 loc[] 或 iloc[] 方法选择需要遍历的列;接着,可以使用 groupby() 函数将其它列按照分类方式分组,并使用 sum() 函数求和。例如:
```python
import pandas as pd
# 创建一个数据框
df = pd.DataFrame({'Class': ['A', 'B', 'C', 'A', 'B', 'C'],
'Score': [80, 90, 85, 70, 95, 75]})
# 对 Class 列进行遍历
for cls in df['Class'].unique():
# 对 Score 列进行分类求和
score_sum = df.loc[df['Class'] == cls, 'Score'].sum()
print(f"Class {cls}: {score_sum}")
```
输出结果为:
```
Class A: 150
Class B: 185
Class C: 160
```
这里先使用 unique() 方法获取 Class 列的唯一值,然后遍历这些值,并根据条件选择出 Score 列中符合要求的行,最后使用 sum() 函数求和。
阅读全文