Когортная тепловая карта retention

Hard matplotlib визуализация Тинькофф

Условие задачи

Дано: массивы cohorts (5 названий когорт), lifetime_months (M0-M4), матрица retention (5x5, значения 0-100%, нули -- данных ещё нет). Постройте тепловую карту когортного retention: нули показываются как пустые ячейки (без цвета). Требования: - Последовательная палитра (например, жёлто-зелёно-синяя) - Аннотация процентом в каждой заполненной ячейке (нули пропускаются) - Цвет текста: белый для значений > 60%, чёрный для остальных - Colorbar, подписи осей - Заголовок "Когортный Retention" Верните объект fig. Сохраните в `result`.

Темы

cohort heatmap retention matplotlib

Подсказки

Что проверяет эта задача

Задача «Когортная тепловая карта retention» учит решать задачи на matplotlib так, как это спрашивают на собеседовании аналитика данных. Такие задачи дают на собеседованиях в Тинькофф и других IT-компаниях. Уровень сложный — проверяет глубину и умение собрать решение из нескольких приёмов.

Как подступиться к решению

Сначала разбери условие и формат входа-выхода, прикинь крайние случаи (пустой ввод, дубли, NaN). Затем выбери структуру данных и собери решение по шагам — начни с простого рабочего варианта, потом оптимизируй сложность. Код пишешь и запускаешь прямо в браузере: Python-тренажёр исполняет его через Pyodide (pandas, numpy, scipy) и проверяет результат автоматически.

На собеседовании по Python для аналитика ценят чистый читаемый код, корректную обработку крайних случаев (пустой ввод, дубли, None) и умение вслух оценить сложность по времени и памяти. Часто просят не просто решить, а разобрать альтернативные подходы и показать, где наивное решение сломается на большом объёме данных.

Прокачать Python для аналитики — бесплатный курс «Pandas с нуля» и Python-тренажёр.

Открыть задачу в тренажёре → ← Все Python-задачи