Python的內(nèi)建模塊itertools提供了非常有用的用于操作迭代對(duì)象的函數(shù)。

首先，我們看看itertools提供的幾個(gè)“無(wú)限”迭代器：

import itertools natuals = itertools.count(1) for n in natuals:
print(n) 1 2 3 ...

因?yàn)閏ount()會(huì)創(chuàng)建一個(gè)無(wú)限的迭代器，所以上述代碼會(huì)打印出自然數(shù)序列，根本停不下來(lái)，只能按Ctrl+C退出。

cycle()會(huì)把傳入的一個(gè)序列無(wú)限重復(fù)下去：

import itertools cs = itertools.cycle('ABC') # 注意字符串也是序列的一種 for c in cs: print(c) 'A' 'B' 'C' 'A' 'B' 'C' ...

同樣停不下來(lái)。

repeat()負(fù)責(zé)把一個(gè)元素?zé)o限重復(fù)下去，不過(guò)如果提供第二個(gè)參數(shù)就可以限定重復(fù)次數(shù)：

ns = itertools.repeat('A', 3) for n in ns: print(n) ... A A A

無(wú)限序列只有在for迭代時(shí)才會(huì)無(wú)限地迭代下去，如果只是創(chuàng)建了一個(gè)迭代對(duì)象，它不會(huì)事先把無(wú)限個(gè)元素生成出來(lái)，事實(shí)上也不可能在內(nèi)存中創(chuàng)建無(wú)限多個(gè)元素。

無(wú)限序列雖然可以無(wú)限迭代下去，但是通常我們會(huì)通過(guò)takewhile()等函數(shù)根據(jù)條件判斷來(lái)截取出一個(gè)有限的序列：

natuals = itertools.count(1) ns = itertools.takewhile(lambda x: x <= 10, natuals) list(ns) [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

itertools提供的幾個(gè)迭代器操作函數(shù)更加有用：

chain()

chain()可以把一組迭代對(duì)象串聯(lián)起來(lái)，形成一個(gè)更大的迭代器：

for c in itertools.chain('ABC', 'XYZ'): print(c) # 迭代效果：'A' 'B' 'C' 'X' 'Y' 'Z'

groupby()

groupby()把迭代器中相鄰的重復(fù)元素挑出來(lái)放在一起：

for key, group in itertools.groupby('AAABBBCCAAA'): print(key, list(group)) . A ['A', 'A', 'A'] B ['B', 'B', 'B'] C ['C', 'C'] A ['A', 'A', 'A']

實(shí)際上挑選規(guī)則是通過(guò)函數(shù)完成的，只要作用于函數(shù)的兩個(gè)元素返回的值相等，這兩個(gè)元素就被認(rèn)為是在一組的，而函數(shù)返回值作為組的key。如果我們要忽略大小寫(xiě)分組，就可以讓元素'A'和'a'都返回相同的key：

for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()): print(key, list(group)) ... A ['A', 'a', 'a'] B ['B', 'B', 'b'] C ['c', 'C'] A ['A', 'A', 'a']

轉(zhuǎn)載于:https://www.cnblogs.com/bruce0425/p/8277746.html

總結(jié)

以上是生活随笔為你收集整理的Python的内建模块itertools的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。