pandasПродвинутоеHardPremium~14 мин

Heatmap-prep: матрица retention для визуализации

// Условие

Таблица df — события пользователей за 2024 год. Нужно построить retention-матрицу для тепловой карты.

Что нужно

Построй retention-матрицу:

Строки: cohort_month — месяц первого визита пользователя (строки '2024-01'...'2024-04');
Колонки: months_since — месяцы с момента регистрации (0..6, 7 значений);
Ячейки: retention rate = nunique(user_id active in this month) / cohort_size.

Column 0 для каждой когорты = 1.0 (все пользователи активны в месяц регистрации). Round до 4 знаков. Sort по index asc.

Верни result — DataFrame с index cohort_month (4 строки) и columns months_since (0..6).

df: event_id, user_id, event_ts (Timestamp)

🔒

Решение, подсказки и автопроверка этой задачи — по подписке Premium. Вместе с ней открываются все премиум-задачи трёх тренажёров: SQL, pandas и A/B.