python中itertools模塊使用小結(jié)

更新時(shí)間：2021年11月04日 08:21:48 作者：嗨，阿良

itertools 是python的迭代器模塊，itertools提供的生成迭代器的函數(shù)，相當(dāng)高效且節(jié)省內(nèi)存。使用這些工具，你將能夠創(chuàng)建自己定制的迭代器用于高效率的循環(huán)。接下來通過本文給大家介紹python中itertools模塊使用，感興趣的朋友一起看看吧

Python 內(nèi)置的 itertools 模塊包含了一系列用來產(chǎn)生不同類型迭代器的函數(shù)或類，這些函數(shù)的返回都是一個(gè)迭代器，我們可以通過 for 循環(huán)來遍歷取值，也可以使用 next() 來取值。

itertools 模塊提供的迭代器函數(shù)有以下幾種類型：

無限迭代器：生成一個(gè)無限序列，比如自然數(shù)序列 1, 2, 3, 4, ...；
有限迭代器：接收一個(gè)或多個(gè)序列（sequence）作為參數(shù)，進(jìn)行組合、分組和過濾等；
組合生成器：序列的排列、組合，求序列的笛卡兒積等。

itertools

高效循環(huán)下創(chuàng)建循環(huán)器的標(biāo)準(zhǔn)庫

Infinite itertools，無限迭代器

itertools.count(start=0, step=10)

默認(rèn)返回一個(gè)從0開始，依次+10的自然數(shù)迭代器，如果你不停止，它會(huì)一直運(yùn)行下去。
可以用start指定開始的位置，step是步長。

import itertools

c = itertools.count(start=0, step=1)
for i in c:
    print(i)

    
# 0
# 10
# 20
# 30
# 40
# 50
# ...

itertools.cycle(iterable)

傳入一個(gè)可迭代對(duì)象，然后無限循環(huán)迭代。

import itertools

# itertools.count()
l = [1,2,3,4,5]
c = itertools.cycle(l)

for i in c:
    print(i)


# 1
# 2
# 3
# 4
# 5
# 1
# 2
# 3
# 4
# 5
# ...

itertools.repeat(p_object, times=None)

重復(fù)迭代一個(gè)對(duì)象p_object，如果不指定times，則會(huì)迭代無數(shù)次，也可以通過times參數(shù)指定迭代的次數(shù)。

import itertools

# itertools.count()
l = [1,2,3,4,5]
c = itertools.repeat(l, 5)

for i in c:
    print(i)


# [1, 2, 3, 4, 5]
# [1, 2, 3, 4, 5]
# [1, 2, 3, 4, 5]
# [1, 2, 3, 4, 5]
# [1, 2, 3, 4, 5]

Iterators terminating on the shortest input sequence

itertools.accumulate(iterable, func)

返回序列的累計(jì)值或者其他二進(jìn)制函數(shù)。

import itertools

# itertools.count()
l = [1,2,3,4,5]
c = itertools.accumulate(l)

print(c)

for i in c:
    print(i)


# 1
# 3
# 6
# 10
# 15

accumulate()仍然返回的是一個(gè)迭代器，傳一個(gè)list，在for循環(huán)中遍歷打印的時(shí)候發(fā)現(xiàn)，它做了累加操作。(迭代第一個(gè)數(shù)，就是前一個(gè)數(shù)的和，迭代到第二個(gè)數(shù)時(shí)，就是前兩個(gè)數(shù)的和，以此類推)

并且做遞加操作時(shí)支持：list, tuple, str, set, dict

傳入的是dict對(duì)象，那么會(huì)累加迭代dict的key：

import itertools

# itertools.count()
d = {'a': 1, 'b': 2, 'c': 3}
c = itertools.accumulate(d)

print(c)

for i in c:
    print(i)


# <itertools.accumulate object at 0x000001F7A0A90E48>
# a
# ab
# abc

itertools.chain(*iterables)

chain()方法中的參數(shù)可以傳入多個(gè)序列，而且只要是序列即可，不限定序列的數(shù)據(jù)類型。

如：迭代list, tuple, str三個(gè)序列

import itertools

l = [1, 2, 3, 4, 5]
t = (1, 2, 3, 4, 5)
s = 'abcdefg'
c = itertools.chain(l, t, s)

print(c)

for i in c:
    print(i)

# <itertools.chain object at 0x0000026E64801448>
# 1
# 2
# 3
# 4
# 5
# 1
# 2
# 3
# 4
# 5
# a
# b
# c
# d
# e
# f
# g

itertools 笛卡爾積

import itertools

d = [
    [{"a": 1}, {"b": 2}], [{"c": 3}, {"d": 4}, {"e": 5}], [{"f": 6}, {"g": 7}]
]

print(*d)

for i in itertools.product(*d):
    print(i)

# [{'a': 1}, {'b': 2}] [{'c': 3}, {'d': 4}, {'e': 5}] [{'f': 6}, {'g': 7}]
# ({'a': 1}, {'c': 3}, {'f': 6})
# ({'a': 1}, {'c': 3}, {'g': 7})
# ({'a': 1}, {'d': 4}, {'f': 6})
# ({'a': 1}, {'d': 4}, {'g': 7})
# ({'a': 1}, {'e': 5}, {'f': 6})
# ({'a': 1}, {'e': 5}, {'g': 7})
# ({'b': 2}, {'c': 3}, {'f': 6})
# ({'b': 2}, {'c': 3}, {'g': 7})
# ({'b': 2}, {'d': 4}, {'f': 6})
# ({'b': 2}, {'d': 4}, {'g': 7})
# ({'b': 2}, {'e': 5}, {'f': 6})
# ({'b': 2}, {'e': 5}, {'g': 7})

到此這篇關(guān)于python中的itertools模塊簡單使用的文章就介紹到這了,更多相關(guān)python itertools模塊使用內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: