37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
# 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)
|