Сумма по группам в Pandas

Если в таблице повторяются метки категории, сумму по каждой из них удобно получить через метод groupby и метод sum. Сначала строки собирают по столбцу с меткой, затем складывают числа в выбранном столбце. На выходе - ряд: подпись группы и её сумма.

Создадим таблицу с категориями и ценами, сгруппируем по столбцу category и сложим столбец price:

import pandas as pd

data = {
	'category': ['A', 'B', 'A', 'B'],
	'price': [10, 20, 15, 25],
}
shop = pd.DataFrame(data)
groups = shop.groupby('category')
totals = groups['price'].sum()
print(totals) # выведет category
              #          A    25
              #          B    45
              # Name: price, dtype: int64

В ряде две строки: для категории A сумма 25, для B - 45. Имя столбца в конце подсказывает, что именно суммировали.

Соберите таблицу с городами в столбце city и населением в столбце pop на четыре строки с двумя городами. Выведите сумму населения по каждому городу.

Создайте таблицу продаж: столбец dept и столбец amount с несколькими отделами. Получите сводку - сумму amount отдельно для каждого отдела - и выведите её.

Из словаря со столбцами team и points соберите таблицу на шесть строк и трёх команд. Выведите сумму баллов по каждой команде.