Сгруппируйте строки по user_id и просуммируйте revenue. Получатся суммы: пользователь 1 — 450, 2 — 550, 3 — 300, 4 — 400.
import pandas as pd
Сгруппируйте строки по user_id и просуммируйте revenue. Получатся суммы: пользователь 1 — 450, 2 — 550, 3 — 300, 4 — 400.
Короткий ответ
Что ответить на собеседовании
Подробный разбор
Ответ с пояснениями
Условие
import pandas as pd import numpy as np
data = { 'user_id': [1, 1, 1, 2, 2, 3, 3, 3, 4], 'date': ['[phone]', '[phone]', '[phone]', '[phone]', '[phone]', '[phone]', '[phone]', '[phone]', '[phone]'], 'revenue': [100, 200, 150, 300, 250, 100, 100, 100, 400] } df = pd.DataFrame(data) df['date'] = pd.to_datetime(df['date']) Посчитать общую сумму выручки (revenue) для каждого пользователя
Ответ
Сгруппируйте строки по user_id и просуммируйте revenue. Получатся суммы: пользователь 1 — 450, 2 — 550, 3 — 300, 4 — 400.
Для этой задачи даты не нужны, поэтому строку pd.to_datetime с повреждёнными значениями [phone] выполнять не следует.
df = pd.DataFrame(data)
result = df.groupby('user_id', as_index=False)['revenue'].sum()
Результат содержит по одной строке на пользователя. Если даты понадобятся для другой задачи, необходимо восстановить исходные значения, а не угадывать их.