Получить доступ

Какие инструменты itertools стоит знать?

MiddleИтераторы и генераторы
Ещё спрашивают так:
  • Как обработать большой поток данных в Python без загрузки в память?
Коротко

Ленивые строительные блоки: chain (склеить), islice (срез итератора), groupby (группы подряд идущих — нужна предварительная сортировка), product/permutations/combinations, accumulate (накопительные суммы), batched (3.12, пачки по n), tee, zip_longest, count/cycle/repeat.

  • Всё лениво: работает с бесконечными и огромными потоками
  • groupby без сортировки разобьёт одинаковые ключи на несколько групп
  • Итератор одноразовый: после прохода пуст

А ещё more-itertools — расширенный набор (chunked, windowed, first)

Так подсказку видит пользователь sotto на живом собеседовании — за пару секунд, пока вы думаете вслух.

Подробно

Зачем. Итераторы обрабатывают элементы по одному, не создавая промежуточных списков: файл на 10 ГБ читается построчно с постоянной памятью.

Частые ловушки:

  • groupby группирует только соседние элементы с одинаковым ключом;
  • группа — итератор, который «исчезает» при переходе к следующей — материализовать list(group), если нужна позже;
  • tee хранит буфер разницы между копиями — при сильном расхождении съедает память;
  • повторный обход исчерпанного итератора молча ничего не даёт.

batched (3.12) — частая задача «отправлять в БД пачками»; в старых версиях — islice в цикле или more_itertools.chunked.

Пример

from itertools import islice, groupby, batched, chain

with open("big.log") as f:
    first_100 = list(islice(f, 100))

rows = sorted(orders, key=lambda o: o.city)
for city, group in groupby(rows, key=lambda o: o.city):
    print(city, sum(o.total for o in group))

for chunk in batched(ids, 500):      # Python 3.12
    db.fetch_many(chunk)

Что хотят услышать

  • Основные функции и ленивость.
  • Ловушка groupby.
  • Применение к большим данным.

Связанные вопросы

Закрытая бета

Подсказки на собеседовании — в реальном времени

sotto слышит вопрос интервьюера и показывает короткий ответ и план, как в блоке «Коротко». Оставьте контакт — пришлём доступ и промокод.

  • Промокод на 2 часа — сверх пробных 30 минут
  • Программа для Windows — скачать сразу, войти — через Яндекс ID или VK ID
  • Прямая связь с разработчиком: что улучшить — сделаем