今天了解了下python中內(nèi)置模塊itertools的使用，熟悉下，看能不能以后少寫(xiě)幾個(gè)for,嘿嘿😁。

1.無(wú)窮的迭代器

1.1 count(start,[step])

count()接受兩個(gè)參數(shù)

start:循環(huán)開(kāi)始的數(shù)字

step:循環(huán)中的間隔

from itertools import count

"""

無(wú)窮的迭代器 count()

"""

c = count(0,2)

v = next(c)

while v < 10:

v = next(c)

print(v,end=',')

1.2 cycle()

cycle就是一while True，無(wú)限循環(huán)里面的數(shù)字。

"""

無(wú)窮迭代器 cycle()

"""

from itertools import cycle

c = cycle('ABCD')

for i in range(10):

print(next(c),')

1.3 repeat(elem,[n])

重復(fù)迭代elem，n次

"""

無(wú)窮迭代器 repeat()

"""

from itertools import repeat

r = repeat(1,3)

for i in range(3):

print(next(r),')

2. 迭代器

2.1 accumulate(p,[func])

使用func的函數(shù)對(duì)迭代對(duì)象p進(jìn)行累積。

"""

迭代器 accumulate()

"""

from itertools import accumulate

test_list = [i for i in range(1,11)]

for i in accumulate(test_list): # 默認(rèn)是operator.add

print(i,')

print()

for i in accumulate(test_list,lambda x,y: x * y): # operator.mul

print(i,')

2.2 chain()

chain()中可以放多個(gè)迭代對(duì)象，然后一一迭代出來(lái)。

"""

迭代器 chain()

"""

from itertools import chain

ch = chain([1,2,3],{4: 4,5: 5},{6,7,8},(9,),[10,[11,12]])

for i in ch:

print(i)

2.3 chain.from_iterable()

跟chain不同的地方在于:

chain: 可以接受多個(gè)迭代對(duì)象

chain.from_iterable():可以接受一個(gè)可以產(chǎn)生迭代對(duì)象的迭代器

"""

迭代器 chain.from_iterable()

"""

def gen_iterables():

for i in range(10):

yield range(i)

for i in chain.from_iterable(gen_iterables()):

print(i)

2.4 compress(data,selectors)

這是就是看下這個(gè)就知道了s是selectors中的元素。

(d[0] if s[0]),(d[1] if s[1]),...

"""

迭代器 compress

"""

from itertools import compress

print(list(compress(['A','B','C','D'],[0,1,1])))

2.5 dropwhile(pred,seq)

循環(huán)開(kāi)始的條件是，直到遇到第一次不滿足pred條件的情況，才開(kāi)始遍歷。

"""

迭代器 dropwhile()

"""

from itertools import dropwhile

l = [1,6,3,8,10]

print(list(dropwhile(lambda x: x < 3,l)))

2.6 groupby

這個(gè)感覺(jué)挺有意思的，有點(diǎn)像sql中的group_by。可以對(duì)字符串，列表等進(jìn)行分組。

返回鍵和，組里的內(nèi)容

from itertools import groupby

# 對(duì)字符串進(jìn)行分組

for k,g in groupby('11111234567'):

print(k,list(g))

d = {1: 1,2: 2,3: 2}

# 按照字典value來(lái)進(jìn)行分組

for k,g in groupby(d,lambda x: d.get(x)):

print(k,list(g))

2.7 islice

這個(gè)就是對(duì)迭代對(duì)象進(jìn)行切割，不支持負(fù)數(shù)，有點(diǎn)像range(1,10,2)這種

from itertools import islice

print(list(islice('ABCDEFG',None)))

2.8 zip_longest

這個(gè)和zip很像，不同地方在于:

zip結(jié)束取決于里面最短的迭代對(duì)象

zip_longest結(jié)束取決于里面最長(zhǎng)的迭代對(duì)象

from itertools import zip_longest

for x,y in zip_longest([1,[1,2]):

print(x,y)

for x,y in zip([1,y)

排列組合迭代器

3.1 product

相當(dāng)于 嵌套的for

“”"

排列組合迭代器 product 嵌套的for

“”"

from itertools import product

for i,j in product([1,[4,5]):

print(i,j

3.2 permutations

全排列，比如輸出123的全部情況。(1,3),(1,2)…

from itertools import permutations

print(list(permutations('123')))

3.3 combinations(p,r)

從p中找出所有長(zhǎng)度為r的排列情況… 有順序

from itertools import combinations

print(list(combinations([1,2)))

3.4 combinations_with_replacement()

從p中找出所有長(zhǎng)度為r的排列情況，有順序，但包括自身就是會(huì)重復(fù)的意思。

combinations_with_replacement(‘ABCD',2)

AA AB AC AD BB BC BD CC CD DD

了解是了解了，就是用的時(shí)候不知道能不能想起來(lái)…

以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持我們。

總結(jié)

以上是生活随笔為你收集整理的python中itertools的用法_python中的itertools的使用详解的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。