前言：

Python的內(nèi)建模塊itertools提供了非常有用的用于操作迭代對(duì)象的函數(shù)，itertools提供的幾個(gè)“無(wú)限”迭代器

1、count()創(chuàng)建自然數(shù)序列

import itertools
?
n = 0
natuals = itertools.count(1)
for m in natuals:
? ? print(m)

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

運(yùn)行結(jié)果：

0 1 2 3 4 5 6 7 ...

2、cycle()創(chuàng)建序列循環(huán)

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

示例1、循環(huán)列表中的內(nèi)容：

natuals = itertools.cycle([1, 2, 3])
for m in natuals:
? ? print(m)

運(yùn)行結(jié)果：

1 2 3 1 2 3 1 2 3 ...

示例2、字符串中字符循環(huán)：

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

運(yùn)行結(jié)果：

A B C A B C A B C ...

3、repeat()把一個(gè)元素有限次重復(fù)

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

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

運(yùn)行結(jié)果：

A A A ...

4、takewhile()根據(jù)條件判斷來(lái)截取出一個(gè)有限的序列

        無(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è)有限的序列。

import itertools
?
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 10, natuals)
print(list(ns))

運(yùn)行結(jié)果：

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

迭代器測(cè)試：

        對(duì)于有限數(shù)據(jù)的迭代器，每調(diào)用一次，迭代器內(nèi)數(shù)據(jù)彈出。

import itertools
?
natuals = itertools.count(1)
ns = itertools.takewhile(lambda x: x <= 3, natuals)
for n in ns:
? ? print(n)
?
print(list(ns))

運(yùn)行結(jié)果：

1 2 3 []

5、chain()迭代對(duì)象串聯(lián)

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

for c in itertools.chain('ABC', 'XYZ'):
? ? print(c)

運(yùn)行結(jié)果：

'A' 'B' 'C' 'X' 'Y' 'Z'

6、groupby()迭代器元素分組

       groupby()把迭代器中相鄰的重復(fù)元素挑出來(lái)放在一起，更多的用于字符串。

for key, group in itertools.groupby('AAABBBCCAAA'):
? ? print(key, list(group))

運(yùn)行結(jié)果：

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。如果我們要忽略大小寫分組，就可以讓元素'A'和'a'都返回相同的key。

for key, group in itertools.groupby('AaaBBbcCAAa', lambda c: c.upper()):
? ? print(key, list(group))

運(yùn)行結(jié)果：

A ['A', 'a', 'a']
B ['B', 'B', 'b']
C ['c', 'C']
A ['A', 'A', 'a']

 到此這篇關(guān)于python itertools包內(nèi)置無(wú)限迭代器的文章就介紹到這了,更多相關(guān)python itertools內(nèi)置迭代器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

最新評(píng)論