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

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

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

一、引入方式

from itertools import cycle

二、使用方法

我們先來看看它的源碼

cycle它接收一個可迭代對象，可以將一個可迭代對象轉換為一個可以無限迭代的迭代器

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

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

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

打印結果：

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

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

from itertools import cycle

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

打印結果：

......

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

接下來我們使用next()進行迭代，看看會發(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))

打印結果：

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

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