快捷導(dǎo)航

python 使用cycle構(gòu)造無限循環(huán)迭代器

更新時(shí)間：2020年12月02日 15:24:45 作者：小公瑾

這篇文章主要介紹了python 使用cycle構(gòu)造無限循環(huán)迭代器的方法，幫助大家更好的理解和學(xué)習(xí)python，感興趣的朋友可以了解下

一、引入方式

from itertools import cycle

二、使用方法

我們先來看看它的源碼

cycle它接收一個(gè)可迭代對(duì)象，可以將一個(gè)可迭代對(duì)象轉(zhuǎn)換為一個(gè)可以無限迭代的迭代器

源碼里我們可以看到它實(shí)現(xiàn)了__iter__和__next__的魔術(shù)方法，說明它既是可迭代對(duì)象也是一個(gè)迭代器，我們可以使用for循環(huán)和next()方法去操作它

我們先來看看以普通的方式去遍歷一個(gè)列表

li = ['python', 'java', 'c', 'ruby', 'php', 'javascript']
for i in li:
  print(i)

打印結(jié)果：

可以看到，遍歷直到最后一個(gè)元素取完后就不再遍歷了

接下來我們使用cycle來創(chuàng)建一個(gè)無限迭代的迭代器

from itertools import cycle

li = ['python', 'java', 'c', 'ruby', 'php', 'javascript']
cycle_iter = cycle(li)
for i in cycle_iter:
  print(i)

打印結(jié)果：

......

我們可以發(fā)現(xiàn)當(dāng)遍歷到最后一個(gè)元素結(jié)束后又會(huì)回到第一個(gè)元素，就這樣無限循環(huán)下去形成了死循環(huán)，跟while死循環(huán)很相似

接下來我們使用next()進(jìn)行迭代，看看會(huì)發(fā)生什么

from itertools import cycle

li = ['python', 'java', 'c', 'ruby', 'php', 'javascript']
cycle_iter = cycle(li)
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))
print(next(cycle_iter))

打印結(jié)果：

我們發(fā)現(xiàn)當(dāng)最后一個(gè)元素迭代完后，又繼續(xù)迭代第一個(gè)元素，這跟普通的迭代器不一樣，普通的迭代器最后一個(gè)元素迭代完后繼續(xù)迭代，會(huì)拋出迭代停止的異常

三、應(yīng)用場景
準(zhǔn)備一批數(shù)據(jù)需要循環(huán)使用，并且需要一定的順序，并不希望隨機(jī)選取一個(gè)使用，這時(shí)cycle就起到了很好的作用，以我工作中為例，需要針對(duì)某知名網(wǎng)站爬取數(shù)據(jù)時(shí)，網(wǎng)站對(duì)你的請(qǐng)求頻率做了限制，而你又需要更高效的獲取這些數(shù)據(jù)，就可以使用cookie池，在循環(huán)發(fā)起請(qǐng)求時(shí)，就可以用next()方法按順序去迭代獲取cookie池中的數(shù)據(jù)用來發(fā)送請(qǐng)求，迭代完最后一個(gè)cookie后，又會(huì)回到第一個(gè)cookie，這樣一來，當(dāng)賬號(hào)比較多時(shí)，每個(gè)賬號(hào)使用到的頻率時(shí)間就不會(huì)太大，也就達(dá)到了預(yù)期效果，部分代碼截圖示例：