具有總和和唯一值的大熊貓分組

Question

我有以下數據框

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()

Answer 1

在groupby上使用agg函數：

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

對於平均值，您可以通過將pandas方法放在引號'mean'或將numpy方式表示為np.mean