python中的itertools的使用詳解
今天瞭解了下python中內建模組itertools的使用,熟悉下,看能不能以後少寫幾個for,嘿嘿😁。
1.無窮的迭代器
1.1 count(start,[step])
count()接受兩個引數
- start:迴圈開始的數字
- step:迴圈中的間隔
from itertools import count """ 無窮的迭代器 count() """ c = count(0,2) v = next(c) while v < 10: v = next(c) print(v,end=',')
1.2 cycle()
cycle就是一while True,無限迴圈裡面的數字。
""" 無窮迭代器 cycle() """ from itertools import cycle c = cycle('ABCD') for i in range(10): print(next(c),')
1.3 repeat(elem,[n])
重複迭代elem,n次
""" 無窮迭代器 repeat() """ from itertools import repeat r = repeat(1,3) for i in range(3): print(next(r),')
2. 迭代器
2.1 accumulate(p,[func])
使用func的函式對迭代物件p進行累積。
""" 迭代器 accumulate() """ from itertools import accumulate test_list = [i for i in range(1,11)] for i in accumulate(test_list): # 預設是operator.add print(i,') print() for i in accumulate(test_list,lambda x,y: x * y): # operator.mul print(i,')
2.2 chain()
chain()中可以放多個迭代物件,然後一一迭代出來。
""" 迭代器 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: 可以接受多個迭代物件
- chain.from_iterable():可以接受一個可以產生迭代物件的迭代器
""" 迭代器 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)
這是就是看下這個就知道了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)
迴圈開始的條件是,直到遇到第一次不滿足pred條件的情況,才開始遍歷。
""" 迭代器 dropwhile() """ from itertools import dropwhile l = [1,6,3,8,10] print(list(dropwhile(lambda x: x < 3,l)))
2.6 groupby
這個感覺挺有意思的,有點像sql中的group_by。可以對字串,列表等進行分組。
返回鍵和,組裡的內容
from itertools import groupby # 對字串進行分組 for k,g in groupby('11111234567'): print(k,list(g)) d = {1: 1,2: 2,3: 2} # 按照字典value來進行分組 for k,g in groupby(d,lambda x: d.get(x)): print(k,list(g))
2.7 islice
這個就是對迭代物件進行切割,不支援負數,有點像range(1,10,2)這種
from itertools import islice print(list(islice('ABCDEFG',None)))
2.8 zip_longest
這個和zip很像,不同地方在於:
- zip結束取決於裡面最短的迭代物件
- zip_longest結束取決於裡面最長的迭代物件
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
相當於 巢狀的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中找出所有長度為r的排列情況… 有順序
from itertools import combinations print(list(combinations([1,2)))
3.4 combinations_with_replacement()
從p中找出所有長度為r的排列情況,有順序,但包括自身就是會重複的意思。
- combinations_with_replacement(‘ABCD',2)
- AA AB AC AD BB BC BD CC CD DD
瞭解是瞭解了,就是用的時候不知道能不能想起來…
以上就是本文的全部內容,希望對大家的學習有所幫助,也希望大家多多支援我們。