這篇文章主要介紹了Python迭代器模組itertools使用原理解析,文中通過示例程式碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值,需要的朋友可以參考下

介紹

今天介紹一個很強大的模組，而且是python自帶的，那就是itertools迭代器模組。

使用

使用起來很簡單，先匯入模組

import itertools

下面，我們通過一些例子邊學邊練

三個無限迭代器

先告訴大家 control + C 可以強制停止程式哦

1.count()

num = itertools.count(10)
for i in num:
    print(i)
# 10
# 11
# 12
# 13
# 以此類推，無窮無盡

2.cycle()

letter = itertools.cycle('ABC')
for i in letter:
    print(i)
# A
# B
# C
# A
# B
# 依次迴圈，無窮無盡

3.repeat()

rp = itertools.repeat('X')
for i in rp:
    print(i)

# X
# X
# X
# 依次類推，無窮無盡

rp2 = itertools.repeat('X',2) # 限制2次
for i in rp2:
    print(i)

# X
# X

想要限制迭代的次數還有一個辦法，就是使用takewhile

num2 = itertools.takewhile(lambda x: x < 15,num)
list(num2)
# [10,11,12,13,14]

可以用來把幾個迭代器合起來，構成一整個迭代器

for c in itertools.chain('AB','CD'):
    print(c)
# A
# B
# C
# D

groupby()

可以把重複的元素group起來

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

# A ['A','A','A']
# B ['B','B','B]
# C ['C','C']

# 注意這裡是區分大小寫的，如果要忽略
# 請使用 itertools.groupby('AAABBCCB',lambda c: c.upper())

accumulate

累加

x = itertools.accumulate(range(5))
print(list(x))
# [0,1,3,6,10,15]

tee

可以將一個迭代器拆分為n個迭代器

a = [1,2,4,5]
x1,x2,x3 = itertools.tee(a,3)

# 產生了三個元素和a一樣的iter

combinations

求列表或生成器中指定數目的元素不重複的所有組合

x = itertools.combinations(range(4),3)
print(list(x))
# [(0,2),(0,3),(1,3)]

compress

按照真值表篩選元素

x = itertools.compress(range(5),(True,False,True,False)) # 0,5
print(list(x))
# [0,3]

filterfalse

保留對應真值為False的元素

x = itertools.filterfalse(lambda n: n < 5,5,7,0))
print(list(x))
# [5,10]

islice

對迭代器進行切片，引數分別是iter，start，end， step

x = itertools.islice(range(10),9,2)
print(list(x))
# [0,8]

product

產生類似笛卡爾積

x = itertools.product('ABC',range(3))
print(list(x))
# [('A',0),('A',1),('B',('C',2)]

zip_longest(*iterables,fillvalue=None)

建立一個迭代器，從每個可迭代物件中收集元素。如果可迭代物件的長度未對齊，將根據 fillvalue 填充缺失值。迭代持續到耗光最長的可迭代物件。

以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支援我們。