Podstawowe itertools

Python oferuje wiele narzędzi do iteracji: niektóre są wbudowane (zawsze dostępne), inne znajdują się w module itertools biblioteki standardowej. Oto najważniejsze z nich.

enumerate: indeks + wartość

Gdy podczas pętli potrzebujesz zarówno indeksu, jak i samego elementu:

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

# rozpoczęcie od innej liczby:
for i, parola in enumerate(["a", "b", "c"], start=1):
    print(i, parola)

zip: iteracja równoległa

Łączy ze sobą wiele obiektów iterowalnych element po elemencie:

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

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

Funkcja zip kończy działanie na najkrótszym z obiektów. Dla dłuższego łączenia użyj itertools.zip_longest.

Tworzenie słownika z dwóch równoległych list:

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

itertools.chain: łączenie obiektów iterowalnych

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

Bez tworzenia pośredniej listy: przydatne dla plików, generatorów itp.

itertools.count i repeat: nieskończone iteratory

from itertools import count, repeat

# count(start, step) — nieskończone rosnące liczby całkowite
for i in count(10, 2):
    if i > 20: break
    print(i)
# 10, 12, 14, 16, 18, 20

# repeat(x, n) — powtarza tę samą wartość n razy (lub nieskończenie bez n)
list(repeat("ok", 3))   # ['ok', 'ok', 'ok']

itertools.combinations

Wszystkie kombinacje (bez powtórzeń, niezależnie od kolejności) o określonej długości:

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

Wariant z uwzględnieniem kolejności: permutations. Warianty z powtórzeniami: combinations_with_replacement oraz product.

zip() dla słowników i niezbalansowanych obiektów iterowalnych

Funkcja zip() przerywa działanie, gdy tylko najkrótszy z obiektów iterowalnych ulegnie wyczerpaniu. Jeśli chcesz połączyć elementy i zachować nadmiarowe wartości (uzupełniając je wartością domyślną, np. None), możesz użyć funkcji zip_longest() z modułu itertools. Dodatkowo, wynik funkcji zip możesz przekazać bezpośrednio do dict(), aby natychmiast utworzyć słownik.

Spróbuj sam

Ćwiczenie powtórzeniowe

Dodatkowe wyzwanie