Python中itertools模塊的使用教程詳解
itertools模塊的介紹
在Python中，迭代器（Iterator）是常用來(lái)做惰性序列的對(duì)象，只有當(dāng)?shù)侥硞€(gè)值的時(shí)候，才會(huì)進(jìn)行計(jì)算得出這個(gè)值。因此，迭代器可以用來(lái)存儲(chǔ)無(wú)限大的序列，這樣我們就不用把他一次性放在內(nèi)存中，而只在需要的時(shí)候進(jìn)行計(jì)算。所以，對(duì)于讀取大文件或者無(wú)線集合，最好是使用迭代器。實(shí)際上，Python2的大多數(shù)函數(shù)都是返回列表等序列，而Python3都已經(jīng)改進(jìn)為返回迭代器。
Python的內(nèi)置模塊itertools就是用來(lái)操作迭代器的一個(gè)模塊，包含的函數(shù)都是能夠創(chuàng)建迭代器來(lái)用于for循環(huán)或者next()。其中函數(shù)主要可以分為三類，分別是無(wú)限迭代器，有限迭代器，組合迭代器。
無(wú)限迭代器（Infinite Iterators）
這些函數(shù)可以生成無(wú)限的迭代器，我們主要學(xué)習(xí)以下三個(gè)函數(shù)的用法。
count([start=0, step=1]) 接收兩個(gè)可選整形參數(shù)，第一個(gè)指定了迭代開始的值，第二個(gè)指定了迭代的步長(zhǎng)。此外，start參數(shù)默認(rèn)為0，step參數(shù)默認(rèn)為1，可以根據(jù)需要來(lái)把這兩個(gè)指定為其它值，或者使用默認(rèn)參數(shù)。
import itertools
for i in itertools.count(10,2):
print(i)
if i>20:
break
[Running] python -u "e:\pythonee\code\test.py"
10
12
14
16
18
20
22cycle(iterable) 是用一個(gè)可迭代對(duì)象中的元素來(lái)創(chuàng)建一個(gè)迭代器，并且復(fù)制自己的值，一直無(wú)限的重復(fù)下去。
import itertools
for i in itertools.cycle("abcd"):
print(i) # 具有無(wú)限的輸出，可以按ctrl+c來(lái)停止。
[Running] python -u "e:\pythonee\code\test.py"
a
b
c
d
a
b
c
d
a
brepeat(elem [,n])是將一個(gè)元素重復(fù)n遍或者無(wú)窮多遍，并返回一個(gè)迭代器。
import itertools
for i in itertools.repeat("abcd",5):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
abcd
abcd
abcd
abcd
abcd組合迭代器（Combinatoric Iterators）
組合操作包括排列，笛卡兒積，或者一些離散元素的選擇，組合迭代器就是產(chǎn)生這樣序列的迭代器。我們來(lái)看看這幾個(gè)函數(shù)的用法。
product(*iterables, repeat=1) 得到的是可迭代對(duì)象的笛卡兒積，*iterables參數(shù)表示需要多個(gè)可迭代對(duì)象。這些可迭代對(duì)象之間的笛卡兒積，也可以使用for循環(huán)來(lái)實(shí)現(xiàn)，例如 product(A, B) 與 ((x,y) for x in A for y in B)就實(shí)現(xiàn)一樣的功能。
import itertools
for i in itertools.product([1,2,3],[4,5,6]):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
(1, 4)
(1, 5)
(1, 6)
(2, 4)
(2, 5)
(2, 6)
(3, 4)
(3, 5)
(3, 6)而 repeat 參數(shù)則表示這些可迭代序列重復(fù)的次數(shù)。例如 product(A, repeat=4) 與 product(A, A, A, A)實(shí)現(xiàn)的功能一樣。
import itertools
for i in itertools.product('ab','cd',repeat = 2):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
('a', 'c', 'a', 'c')
('a', 'c', 'a', 'd')
('a', 'c', 'b', 'c')
('a', 'c', 'b', 'd')
('a', 'd', 'a', 'c')
('a', 'd', 'a', 'd')
('a', 'd', 'b', 'c')
('a', 'd', 'b', 'd')
('b', 'c', 'a', 'c')
('b', 'c', 'a', 'd')
('b', 'c', 'b', 'c')
('b', 'c', 'b', 'd')
('b', 'd', 'a', 'c')
('b', 'd', 'a', 'd')
('b', 'd', 'b', 'c')
('b', 'd', 'b', 'd')permutations(iterable,r=None)返回的是可迭代元素中的一個(gè)排列組合，并且是按順序返回的，且不包含重復(fù)的結(jié)果。
import itertools
for i in itertools.permutations('abc'):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
('a', 'b', 'c')
('a', 'c', 'b')
('b', 'a', 'c')
('b', 'c', 'a')
('c', 'a', 'b')
('c', 'b', 'a')當(dāng)然，第 2 個(gè)參數(shù)默認(rèn)為None，它表示的是返回元組（tuple) 的長(zhǎng)度，我們來(lái)嘗試一下傳入第二個(gè)參數(shù)。
import itertools
for i in itertools.permutations('abc',2):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
('a', 'b')
('a', 'c')
('b', 'a')
('b', 'c')
('c', 'a')
('c', 'b')combinations(iterable,r) 返回的是可迭代對(duì)象所有的長(zhǎng)度為 r 的子序列，注意這與前一個(gè)函數(shù) permutation 不同，permutation 返回的是排列，而 combinations 返回的是組合。
import itertools
for i in itertools.combinations('1234',2):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
('1', '2')
('1', '3')
('1', '4')
('2', '3')
('2', '4')
('3', '4')combinations_with_replacement(iterable, r) 返回一個(gè)可與自身重復(fù)的元素組合，用法類似于 combinations 。
import itertools
for i in itertools.combinations_with_replacement('1234',2):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
('1', '1')
('1', '2')
('1', '3')
('1', '4')
('2', '2')
('2', '3')
('2', '4')
('3', '3')
('3', '4')
('4', '4')有限迭代器（Iterators Terminating on the Shortest Input Sequence）
這里的函數(shù)有十來(lái)個(gè)，主要為大家介紹其中幾個(gè)常用的函數(shù)。
chain(*iterables) 可以把多個(gè)可迭代對(duì)象組合起來(lái)，形成一個(gè)更大的迭代器。
import itertools
for i in itertools.chain('good','bye'):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
g
o
o
d
b
y
egroupby(iterable,key=None) 可以把相鄰元素按照 key 函數(shù)分組，并返回相應(yīng)的 key 和 groupby，如果key函數(shù)為 None，則只有相同的元素才能放在一組。
import itertools
for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
print(list(group))
[Running] python -u "e:\pythonee\code\test.py"
['A', 'a', 'a']
['B', 'B', 'b']
['c', 'C']
['A', 'A', 'a']accumulate(iterable [,func]) 可以計(jì)算出一個(gè)迭代器，這個(gè)迭代器是由特定的二元函數(shù)的累計(jì)結(jié)果生成的，如果不指定的話，默認(rèn)函數(shù)為求和函數(shù)。
import itertools
for i in itertools.accumulate([0,1,0,1,1,2,3,5]):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
0
1
1
2
3
5
8
13如果我們指定這個(gè)累計(jì)函數(shù)，則還能有不同的用法，例如，指定一個(gè)最大值函數(shù)，或者自己定義的函數(shù)。
import itertools
for i in itertools.accumulate([2,1,4,3,5],max):
print(i)
[Running] python -u "e:\pythonee\code\test.py"
2
2
4
4
5到此這篇關(guān)于Python中itertools模塊的使用教程詳解的文章就介紹到這了,更多相關(guān)Python itertools模塊內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
本篇博客將引領(lǐng)讀者穿越數(shù)學(xué)、計(jì)算和可視化的領(lǐng)域,通過(guò)豐富的示例代碼,揭示π的獨(dú)特之處,無(wú)論是計(jì)算π的各種方法、應(yīng)用領(lǐng)域中的角色,還是π作為無(wú)理數(shù)的特性,我們將通過(guò)Python的鏡頭,發(fā)現(xiàn)這個(gè)數(shù)字在數(shù)學(xué)世界中的非凡之處2024-01-01
這篇文章主要介紹了剖析Python的Twisted框架的核心特性,包括Twisted框架中reacter與Deferred等的關(guān)鍵用法,需要的朋友可以參考下2016-05-05
Python3.0中普通方法、類方法和靜態(tài)方法的比較
Python統(tǒng)計(jì)詞頻的幾種方法小結(jié)
利用Python過(guò)濾相似文本的簡(jiǎn)單方法示例