Код Junior Производительность и память спрашивают часто обновлено
Категории сумм через np.select
Условие
Дан список сумм amounts. Верните список категорий: "low" для суммы меньше 100, "mid" от 100 до 999.99... (меньше 1000), "high" для 1000 и больше. Решите векторно, без цикла и apply.
Примеры
tier([50, 100, 999, 1000, 5000]) // ["low", "mid", "mid", "high", "high"]
tier([]) // []
tier([0]) // ["low"]
Как подойти
Подсказки: 1
np.select([условие1, условие2], [значение1, значение2], default=...)
Решите в тренажёре
Редактор с автотестами прямо в браузере. Можно решить спокойно, вслух или под таймером, как на собесе.
Термины по теме
Ещё по теме «Производительность и память»
- Чем заменить apply(axis=1)
- Когда помогает тип category
- Цепочное присваивание в pandas 3.0
- Обнулить тестовые заказы через loc
- category теряется при concat
- Copy-on-Write в pandas 3.0
- Copy-on-Write: изменение колонки
- Copy-on-Write и SettingWithCopyWarning