今天了解了下python中內置模塊itertools的使用，熟悉下，看能不能以后少寫幾個for,嘿嘿。
1.無窮的迭代器
1.1 count(start,[step])
count()接受兩個參數(shù)
- start:循環(huán)開始的數(shù)字
- step:循環(huán)中的間隔
|
1
2
3
4
5
6
7
8
9
10
|
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，無限循環(huán)里面的數(shù)字。
|
1
2
3
4
5
6
7
8
|
"""無窮迭代器 cycle()"""from itertools import cyclec = cycle('ABCD')for i in range(10): print(next(c), end=',') |
1.3 repeat(elem,[n])
重復迭代elem，n次
|
1
2
3
4
5
6
7
8
|
"""無窮迭代器 repeat()"""from itertools import repeatr = repeat(1, 3)for i in range(3): print(next(r), end=',') |
2. 迭代器
2.1 accumulate(p,[func])
使用func的函數(shù)對迭代對象p進行累積。
|
1
2
3
4
5
6
7
8
9
10
11
|
"""迭代器 accumulate()"""from itertools import accumulatetest_list = [i for i in range(1, 11)]for i in accumulate(test_list): # 默認是operator.add print(i, end=',')print()for i in accumulate(test_list, lambda x, y: x * y): # operator.mul print(i, end=',') |
2.2 chain()
chain()中可以放多個迭代對象，然后一一迭代出來。
|
1
2
3
4
5
6
7
8
|
"""迭代器 chain()"""from itertools import chainch = 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():可以接受一個可以產(chǎn)生迭代對象的迭代器
|
1
2
3
4
5
6
7
8
9
|
"""迭代器 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]), ...
|
1
2
3
4
5
6
|
"""迭代器 compress"""from itertools import compressprint(list(compress(['A', 'B', 'C', 'D'], [0, 1, 1, 1]))) |
2.5 dropwhile(pred,seq)
循環(huán)開始的條件是，直到遇到第一次不滿足pred條件的情況，才開始遍歷。
|
1
2
3
4
5
6
7
|
"""迭代器 dropwhile()"""from itertools import dropwhilel = [1, 7, 6, 3, 8, 2, 10]print(list(dropwhile(lambda x: x < 3, l))) |
2.6 groupby
這個感覺挺有意思的，有點像sql中的group_by。可以對字符串，列表等進行分組。
返回鍵和，組里的內容
|
1
2
3
4
5
6
7
8
9
|
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
這個就是對迭代對象進行切割，不支持負數(shù)，有點像range(1,10,2)這種
|
1
2
|
from itertools import isliceprint(list(islice('ABCDEFG', 2,3, None))) |
2.8 zip_longest
這個和zip很像，不同地方在于:
- zip結束取決于里面最短的迭代對象
- zip_longest結束取決于里面最長的迭代對象
|
1
2
3
4
5
6
7
|
from itertools import zip_longestfor x,y in zip_longest([1,2,3],[1,2]): print(x,y)for x,y in zip([1,2,3],[1,2]): print(x,y) |
排列組合迭代器
3.1 product
相當于 嵌套的for
|
1
2
3
4
5
6
|
“”"排列組合迭代器 product 嵌套的for“”"from itertools import productfor i,j in product([1,2,3],[4,5]):print(i,j |
3.2 permutations
全排列，比如輸出123的全部情況。(1,2,3),(1,3,2)…
|
1
2
3
|
from itertools import permutationsprint(list(permutations('123'))) |
3.3 combinations(p,r)
從p中找出所有長度為r的排列情況… 有順序
|
1
2
|
from itertools import combinationsprint(list(combinations([1,2,3],2))) |
3.4 combinations_with_replacement()
從p中找出所有長度為r的排列情況，有順序，但包括自身就是會重復的意思。
- combinations_with_replacement(‘ABCD', 2)
- AA AB AC AD BB BC BD CC CD DD
了解是了解了，就是用的時候不知道能不能想起來…
以上就是本文的全部內容，希望對大家的學習有所幫助，也希望大家多多支持服務器之家。
原文鏈接：https://blog.csdn.net/qq_36659627/article/details/103926495
