Агрегат по кускам через chunksize: задача с собеседования

Условие

Дан текст CSV с колонками region,amount,comment и размер куска. Посчитайте сумму amount по region, читая файл кусками chunksize и только нужные колонки (usecols), а частичные результаты объедините. Верните словарь {region: сумма} (int). Для файла без строк {}.

Примеры

sum_by_chunks("region,amount,comment\nN,10,x\nS,5,y\nN,7,z\nS,1,w\nE,2,v\n", 2) // {"E":2,"N":17,"S":6}
sum_by_chunks("region,amount,comment\nN,1,a\n", 10) // {"N":1}
sum_by_chunks("region,amount,comment\n", 3) // {}

Как подойти

Подсказки: 2

for chunk in pd.read_csv(..., chunksize=n, usecols=[...])
Частичные суммы: pd.concat(parts).groupby(level=0).sum()

Решите в тренажёре

Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.

Решить в тренажёре

Предыдущий вопрос

Файл больше памяти в pandas

Следующий вопрос

Вложенный JSON заказов в таблицу

Урок по теме

Чтение и запись данных в pandas

Термины по теме

Int64 с большой буквы

Ещё по теме «Чтение и запись данных»

Все вопросы модуля