Python迭代器模塊itertools使用原理解析

發(fā)布時間：2020-08-26 16:03:16 來源：腳本之家閱讀：177 作者：MrDoghead 欄目：開發(fā)技術(shù)

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

介紹

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

使用

使用起來很簡單，先導(dǎo)入模塊

import itertools

下面，我們通過一些例子邊學(xué)邊練

三個無限迭代器

先告訴大家 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
# 依次循環(huán)，無窮無盡

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

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

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

可以用來把幾個迭代器合起來，構(gòu)成一整個迭代器

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

groupby()

可以把重復(fù)的元素group起來

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

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

# 注意這里是區(qū)分大小寫的，如果要忽略
# 請使用 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,3,4,5]
x1, x2, x3 = itertools.tee(a,3)

# 產(chǎn)生了三個元素和a一樣的iter

combinations

求列表或生成器中指定數(shù)目的元素不重復(fù)的所有組合

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

compress

按照真值表篩選元素

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

filterfalse

保留對應(yīng)真值為False的元素

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

islice

對迭代器進行切片，參數(shù)分別是iter，start，end， step

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

product

產(chǎn)生類似笛卡爾積

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

zip_longest(*iterables, fillvalue=None)

創(chuàng)建一個迭代器，從每個可迭代對象中收集元素。如果可迭代對象的長度未對齊，將根據(jù) fillvalue 填充缺失值。迭代持續(xù)到耗光最長的可迭代對象。

以上就是本文的全部內(nèi)容，希望對大家的學(xué)習(xí)有所幫助，也希望大家多多支持億速云。

向AI問一下細(xì)節(jié)

Python迭代器模塊itertools使用原理解析

猜你喜歡

最新資訊

相關(guān)推薦

相關(guān)標(biāo)簽