Основные инструменты 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(), чтобы мгновенно построить карту значений ключа.

Попробуйте это

Обзор упражнения

Дополнительная задача