Основні інструменти itertools

Python надає багато інструментів ітерації: деякі є вбудованими (завжди доступні), інші живуть у модулі itertools стандарту бібліотека. Ось найважливіші з них.

enumerate: індекс + значення

Коли вам потрібні як індекс, так і елемент під час циклу:

for i, parola in enumerate(["a", "b", "c"]):
    print(i, parola)
# 0 a
# 1 b
# 2 c

# parte da un altro numero:
for i, parola in enumerate(["a", "b", "c"], start=1):
    print(i, parola)

zip: повторити паралельно

Поєднує кілька ітерованих елементів поелементно:

nomi = ["Ada", "Linus", "Grace"]
eta = [36, 54, 85]

for n, e in zip(nomi, eta):
    print(f"{n} ha {e} anni")

zip зупиняється на найкоротшій ітерації. Для «довгого» використання itertools.zip_longest.

Створення dict з двох паралельних списків:

dict(zip(nomi, eta))
# {'Ada': 36, 'Linus': 54, 'Grace': 85}

itertools.chain: конкатенація ітерованих елементів

from itertools import chain
list(chain([1, 2], [3, 4], [5]))
# [1, 2, 3, 4, 5]

Без створення проміжного списку: корисно для файлів, генераторів тощо.

itertools.count та repeat: нескінченні ітератори

from itertools import count, repeat

# count(start, step) — interi crescenti infiniti
for i in count(10, 2):
    if i > 20: break
    print(i)
# 10, 12, 14, 16, 18, 20

# repeat(x, n) — ripete lo stesso valore n volte (o all'infinito senza n)
list(repeat("ok", 3))   # ['ok', 'ok', 'ok']

itertools.combinations

Усі комбінації (без повторень, незалежні від порядку) даного довжина:

from itertools import combinations
list(combinations([1, 2, 3, 4], 2))
# [(1,2), (1,3), (1,4), (2,3), (2,4), (3,4)]

Замовлений варіант: permutations. Варіанти з повторенням: combinations_with_replacement та product.

zip() для словників і незбалансованих ітерацій

Функція zip() зупиняється, щойно вичерпується найкоротша ітерація введення. Якщо ви хочете об’єднати елементи в пари та зберегти будь-які надлишкові значення (заповнивши їх значенням за замовчуванням, наприклад None), ви можете використовувати zip_longest() з модуля itertools. Крім того, ви можете передати результат zip безпосередньо до dict(), щоб миттєво створити карту ключ-значення.

Спробуйте

Повторна вправа

Додатковий виклик