Есть ли способ включить все столбцы в группу для аналогичных данных?

#python-3.x #pandas #dataframe

Вопрос:

Я проводил групповую проверку данных о продажах, которая выглядит как приведенный ниже фрейм данных, чтобы рассчитать сумму количеств и затрат.

Продукт Количество Стоимость
A 12 50
A 12 50
A 12 50
A 12 70
A 12 50

Однако df.loc[:, ['Product', 'Quantity', 'Cost']].groupby(['Product']).sum()) группировка приводит к

Стоимость
Продукт
A 270

Есть ли способ также включить сумму количеств для сценария?

Ответ №1:

Вы можете использовать .agg() , где вы указываете, как вы хотите агрегировать разные столбцы:

 print(df.groupby("Product").agg({"Quantity": "sum", "Cost": "sum"}))
 

С принтами:

          Quantity  Cost
Product                
A              60   270