具有总和和唯一值的大熊猫分组

时间:2018-07-04 03:32:44

标签: python python-3.x pandas dataframe pandas-groupby

我有以下数据框

amount = pd.Series(np.random.randn(10),name='amount')
df = pd.DataFrame({'amount': amount,'price':50})
df['month'] = ['jan','feb','mar','apr','apr','apr','may','may','jun','jun']
df.set_index('month')

我想按“月”分组,对“金额”列求和,并在每个月旁边显示价格。我们可以假设价格在每个月内都是统一的,因此使用mean()函数可以很好地工作。

我不确定如何将sum()和mean()函数结合在一起,尝试将其分为2个函数似乎会导致信息丢失,例如:

df.groupby('month')[['amount']].sum()

1 个答案:

答案 0 :(得分:0)

agg上使用groupby函数:

df.groupby('month').agg({
    'amount': sum,
    'price': 'mean'
})

对于平均值,您可以将pandas方法放在引号'mean'中,或者将numpy表示为np.mean