Сообщить об ошибке.

Модуль itertools в Python, эффективные итераторы для циклов

Готовые итераторы на все случаи для эффективных циклов

Модуль itertools реализует ряд итераторов, основанных на конструкциях из языков программирования APL, Haskell и SML. Каждый был переделан в форму, подходящую для Python.

Модуль стандартизирует основной набор быстрых и эффективных по памяти инструментов, которые полезны сами по себе или в сочетании. Вместе они образуют "алгебру итераторов", позволяющую быстро и эффективно создавать специализированные инструменты в чистом Python.

Например язык SML предоставляет инструмент табулирования: tabulate(f), который генерирует последовательность f(0), f(1), .... Тот же эффект может быть достигнут в Python путем объединения функций map() и itertools.count() в форму map(f, count()).

Эти инструменты и их встроенные аналоги также хорошо работают с высокоскоростными функциями в модуле operator. Например, оператор умножения может отображаться на два вектора для формирования эффективного точечного произведения: sum(map(operator.mul, vector1, vector2)).

Также читайте "Введение в модуль itertools Python."

Бесконечные итераторы:

ИтераторАргументыРезультатПример
count()start, [step]start, start+step, start+2*step, …count(10) --> 10 11 12 13 14 ...
cycle()pp0, p1, … plast, p0, p1, …cycle('ABCD') --> A B C D A B C D ...
repeat()elem [,n]elem, elem, elem, … endlessly or up to n timesrepeat(10, 3) --> 10 10 10

Итераторы, оканчивающиеся на самой короткой входной последовательности:

ИтераторАргументыРезультатПример
accumulate()p [,func]p0, p0+p1, p0+p1+p2, …accumulate([1,2,3,4,5]) --> 1 3 6 10 15
chain()p, q, …p0, p1, … plast, q0, q1, …chain('ABC', 'DEF') --> A B C D E F
chain.from_iterable()iterablep0, p1, … plast, q0, q1, …chain.from_iterable(['ABC', 'DEF']) --> A B C D E F
compress()data, selectors(d[0] if s[0]), (d[1] if s[1]), …compress('ABCDEF', [1,0,1,0,1,1]) --> A C E F
dropwhile()pred, seqseq[n], seq[n+1], starting when pred failsdropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
filterfalse()pred, seqelements of seq where pred(elem) is falsefilterfalse(lambda x: x%2, range(10)) --> 0 2 4 6 8
groupby()iterable[, key]sub-iterators grouped by value of key(v)
islice()seq, [start,] stop [, step]elements from seq[start:stop:step]islice('ABCDEFG', 2, None) --> C D E F G
starmap()func, seqfunc(*seq[0]), func(*seq[1]), …starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000
takewhile()pred, seqseq[0], seq[1], until pred failstakewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
tee()it, nit1, it2, … itn splits one iterator into n
zip_longest()p, q, …(p[0], q[0]), (p[1], q[1]), …zip_longest('ABCD', 'xy', fillvalue='-') --> Ax By C- D-

Комбинаторные итераторы:

IteratorArgumentsResults
product()p, q, … [repeat=1]Декартово произведение, эквивалентное вложенному циклу for
permutations()p[, r]Кортежи r-длины, все возможные упорядочения, без повторяющихся элементов
combinations()p, rКортежи r-длины, в отсортированном порядке, без повторяющихся элементов
combinations_with_replacement()p, rКортежи r-длины, в отсортированном порядке, с повторяющимися элементами