itertools提供的各類迭代器擴展

1. 引用

import itertools
或者
from itertools import *

2. 迭代器

3. 具體說明

I. itertools.repeat(object[, times])

????創(chuàng)建一個迭代器，重復(fù)生成object，times（如果已提供）指定重復(fù)計數(shù)，如果未提供times，將無止盡返回該對象。

from itertools import *

for i in repeat('test', 5):
    print(i)

-----------------------------
test
test
test
test
test

II. itertools.chain(*iterables)

????將多個迭代器作為參數(shù), 但只返回單個迭代器, 返回他們中的每個單一元素

from itertools import *

for i in chain([1, 2, 3], ['a', 'b', 'c']):
    print i
----------------------------------
1
2
3
a
b
c

III. chain.from_iterable(*iterables)

????構(gòu)建類似 chain() 迭代器的另一個選擇。不同的是這里可以以整體形式傳入?yún)?shù)(列表或其他可迭代對象)

print([data for data in chain.from_iterable(['ABC', 'DEF'])])
------------------------
['A', 'B', 'C', 'D', 'E', 'F']

IV. itertools.compress(data, selectors)

????提供一個選擇列表，對原始數(shù)據(jù)進行篩選,根據(jù)后面的0或者1判斷True和False

print([data for data in compress('ABCDEF', [1, 0, 1, 0, 1, 1])])
-------------------------------
['A', 'C', 'E', 'F']

V. itertools.groupby(iterable[, key])

返回一個產(chǎn)生按照key進行分組后的值集合的迭代器.

• 按照keyfunc函數(shù)對序列每個元素執(zhí)行后的結(jié)果分組(每個分組是一個迭代器), 返回這些分組的迭代器

from itertools import groupby
qs = [{'date' : 1},{'date' : 2}]
[(name, list(group)) for name, group in itertools.groupby(qs, lambda p:p['date'])]

Out[77]: [(1, [{'date': 1}]), (2, [{'date': 2}])]


>>> from itertools import *
>>> a = ['aa', 'ab', 'abc', 'bcd', 'abcde']
>>> for i, k in groupby(a, len):
...     print i, list(k)
...
2 ['aa', 'ab']
3 ['abc', 'bcd']
5 ['abcde']

VI. itertools.filter(predicate, iterable)

????返回只包括當(dāng)測試函數(shù)返回true時的項.創(chuàng)建一個迭代器，僅生成iterable中predicate(item)為True的項，如果predicate為None，將返回iterable中所有計算為True的項

對函數(shù)func執(zhí)行返回真的元素的迭代器

from itertools import *

    for i in filter(lambda x: x < 1, [-1, 0, 1, 2, 3, 4, 1, -2]):
        print('Yielding:', i)
--------------------------------
Yielding: -1
Yielding: 0
Yielding: -2

VII. itertools.filterfalse(predicate, iterable)

????和ifilter(函數(shù)相反 ， 返回一個包含那些測試函數(shù)返回false的項的迭代器)
,創(chuàng)建一個迭代器，僅生成iterable中predicate(item)為False的項，如果predicate為None，則返回iterable中所有計算為False的項 對函數(shù)func執(zhí)行返回假的元素的迭代器

from itertools import *

    for i in filterfalse(lambda x: x < 1, [-1, 0, 1, 2, 3, 4, 1, -2]):
        print('Yielding:', i)
-----------------------------------------
    for i in filterfalse(lambda x: x < 1, [-1, 0, 1, 2, 3, 4, 1, -2]):
        print('Yielding:', i)

VIII. itertools.islice(iterable, start, stop[, step])

????根據(jù)索引對迭代器切片，terable[start : stop : step]，將跳過前start個項，迭代在stop所指定的位置停止，step指定用于跳過項的步幅。 與切片不同，負(fù)值不會用于任何start，stop和step， 如果省略了start，迭代將從0開始，如果省略了step，步幅將采用1.

• 返回序列seq的從start開始到stop結(jié)束的步長為step的元素的迭代器

from itertools import *

print 'Stop at 5:'
for i in islice(count(), 5):
    print i

print 'Start at 5, Stop at 10:'
for i in islice(count(), 5, 10):
    print i

print 'By tens to 100:'
for i in islice(count(), 0, 100, 10):
    print i

Stop at 5:
0
1
2
3
4
Start at 5, Stop at 10:
5
6
7
8
9
By tens to 100:
0
10
20
30
40
50
60
70
80
90

IX. itertools.permutations(iterable[, r])

????創(chuàng)建一個迭代器，返回iterable中所有長度為r的項目序列，如果省略了r，那么序列的長度與iterable中的項目數(shù)量相同

• 返回以r為長度的所有隨機組合的排列

 print([data for data in permutations('ABCD', 2)])
---------------------------------------------------
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'A'), ('B', 'C'), ('B', 'D'), ('C', 'A'), ('C', 'B'), ('C', 'D'), ('D', 'A'), ('D', 'B'), ('D', 'C')]

X. itertools.combinations(iterable, r)

????創(chuàng)建一個迭代器，返回iterable中所有長度為r的子序列，返回的子序列中的項按輸入iterable中的順序排序 (不帶重復(fù))

print([data for data in combinations('ABCD', 2)])
------------------------------------------------
[('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'C'), ('B', 'D'), ('C', 'D')]

XI. itertools.combinations_with_replacement(iterable, r)

????創(chuàng)建一個迭代器，返回iterable中所有長度為r的子序列，返回的子序列中的項按輸入iterable中的順序排序 (帶重復(fù))

print([data for data in combinations_with_replacement('ABCD', 2)])
---------------------------------------------
[('A', 'A'), ('A', 'B'), ('A', 'C'), ('A', 'D'), ('B', 'B'), ('B', 'C'), ('B', 'D'), ('C', 'C'), ('C', 'D'), ('D', 'D')]

• itertools中實現(xiàn)了很多有用的迭代器，像這里所展示的都是比較實用的，避免自己重新實現(xiàn)復(fù)雜的代碼
• 更多內(nèi)容參考文檔