python中itertools的用法_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
了解是了解了，就是用的時候不知道能不能想起來…
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持我們。
總結
以上是生活随笔為你收集整理的python中itertools的用法_python中的itertools的使用详解的全部內容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： python中二分查找什么意思_pyth
- 下一篇： websocket python爬虫_p