1 模塊簡(jiǎn)介

Python提供了itertools模塊,可以創(chuàng)建屬于自己的迭代器。itertools提供的工具快速并且節(jié)約內(nèi)存。開(kāi)發(fā)者可以使用這些工具創(chuàng)建屬于自己特定的迭代器,這些特定的迭代器可以用于有效的遍歷。

2 模塊使用

2.1 無(wú)限迭代器

itertools中有個(gè)三個(gè)迭代器是無(wú)限迭代的,這就意味著當(dāng)你在使用它們時(shí),你需要了解你要么從這些迭代器中終止,要么就是無(wú)限循環(huán)。

count

count(start = 0, step = 1),count迭代器返回一系列值,以傳入的start參數(shù)開(kāi)始,Count也可以接受step參數(shù)。

from itertools import count for i in count(10): if i > 20: break else: print i,

通過(guò)條件判斷,如果超出20,就從for循環(huán)中break出來(lái),否則,就打印迭代器中的值,控制臺(tái)輸出,

10 11 12 13 14 15 16 17 18 19 20

另一種限制無(wú)限迭代器的輸出是通過(guò)itertools中的islice方法,如下所示,

from itertools import count,islice for i in islice(count(10),
        		

網(wǎng)友評(píng)論