Панды группового с суммой и уникальными значениями

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 (), и попытка разделить на две функции, похоже, приведет к потере информации, например ::

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

Answer 1

Используйте функцию agg на groupby:

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

Значит, вы можете использовать метод панд, заключив его в кавычки 'mean' или просто обозначая как np.mean