1
下面是摘录:大熊猫如何聚合多个列
test = pd.DataFrame({'userid': [1,1,1,2,2], 'order_id': [1,2,3,4,5], 'fee': [2,1,5,3,1]})
我想组基于用户ID和计数“的order_id”列,总结了“费用”列:
test.groupby('userid').order_id.count()
test.groupby('userid').fee.sum()
是否有可能在一行代码中执行这两个操作,这样我可以得到一个导致DF是这样的:
userid counts sum
...
我已经试过pivot_table:
test.pivot_table(index='userid', values=['order_id', 'fee'], aggfunc=[np.size, np.sum])
它给这样的事情:
size sum
fee order_id fee order_id
userid
1 3 3 8 6
2 2 2 4 9
是否可以告诉熊猫使用np.size & np.sum在一列而不是两个?