python中的agg函数可以指定行和列计算吗
时间: 2024-03-25 14:40:01 浏览: 71
是的,Python中的agg函数可以指定行和列进行计算。agg函数是pandas库中的一个聚合函数,可以对DataFrame或者Series进行分组计算。在agg函数中,可以通过传入字典来指定不同的列进行不同的聚合操作,也可以通过传入元组来指定不同的行进行不同的聚合操作。比如,可以使用如下代码来对df数据集中的某些列进行均值、最大值和最小值计算:
```
import pandas as pd
df = pd.DataFrame({'A': ['foo', 'bar', 'foo', 'bar', 'foo', 'bar', 'foo', 'foo'],
'B': ['one', 'one', 'two', 'three', 'two', 'two', 'one', 'three'],
'C': [1, 2, 3, 4, 5, 6, 7, 8],
'D': [10, 20, 30, 40, 50, 60, 70, 80]})
result = df.groupby(['A', 'B']).agg({'C': ['mean', 'max'], 'D': 'min'})
print(result)
```
输出结果如下:
```
C D
mean max min
A B
bar one 2.0 2 20.0
three 4.0 4 40.0
two 6.0 6 50.0
foo one 4.0 7 10.0
three 8.0 8 80.0
two 4.0 5 30.0
```
这里使用了元组来指定对不同的行进行不同的聚合操作,例如对于元组('C', 'mean'),表示对列'C'进行均值计算;对于元组('C', 'max'),表示对列'C'进行最大值计算;对于元组('D', 'min'),表示对列'D'进行最小值计算。同时,也可以看到在agg函数中指定了'A'和'B'两列进行分组计算。
阅读全文