commit 6a41d2963c4c298e9ace508434e101d9fbbb9a98 Author: Глинский Date: Tue Jan 20 23:49:32 2026 +0000 Загрузить файлы в «/» diff --git a/count_sum_pandas.py b/count_sum_pandas.py new file mode 100644 index 0000000..e828bdc --- /dev/null +++ b/count_sum_pandas.py @@ -0,0 +1,36 @@ +# count_sum_pandas.py + +import pandas as pd + +def count_sum(df: pd.DataFrame) -> pd.DataFrame: + """ + Функция суммирует количество по каждому продукту. + + Параметры: + df : pd.DataFrame + Входной DataFrame с колонками 'Дата', 'Товар', 'Количество' + + Возвращает: + pd.DataFrame + DataFrame с суммой по каждому товару. + Колонка 'Товар' становится индексом. + """ + # Группируем по столбцу 'Товар' и суммируем 'Количество' + result = df.groupby('Товар', as_index=True)['Количество'].sum().to_frame() + + return result + + +# Пример использования +if __name__ == "__main__": + # Пример данных + data = { + 'Дата': ['07.05.2022', '07.05.2022', '08.05.2022', '08.05.2022'], + 'Товар': ['Банан', 'Хлеб', 'Банан', 'Хлеб'], + 'Количество': [30, 10, 40, 8] + } + df_input = pd.DataFrame(data) + + # Вызов функции + df_output = count_sum(df_input) + print(df_output)