python itertools功能詳解

2021-10-01 17:16:57 字數 3643 閱讀 7525

​ 使用只需簡單一句匯入：import itertools

​ 與其名稱意義一樣，給它乙個列表如 lists/tuples/iterables，鏈結在一起；返回iterables物件。

letters = ['a', 'b', 'c', 'd', 'e', 'f']


booleans = [1, 0, 1, 0, 0, 1]


print(list(itertools.chain(letters,booleans)))


#     ['a', 'b', 'c', 'd', 'e', 'f', 1, 0, 1, 0, 0, 1]


print(tuple(itertools.chain(letters,letters[3:])))


#     ('a', 'b', 'c', 'd', 'e', 'f', 'd', 'e', 'f')


print(set(itertools.chain(letters,letters[3:])))


#     


print(list(itertools.chain(letters,letters[3:])))


#     ['a', 'b', 'c', 'd', 'e', 'f', 'd', 'e', 'f']


for item in list(itertools.chain(letters,booleans)):


print(item)
​ 生成無界限序列，count(start=0, step=1) ，示例從100開始，步長為2，迴圈10，列印對應值；必須手動break，count()會一直迴圈。

i = 0


for item in itertools.count(100,2):


i += 1


if i > 10 : break


print(item)
​ python filte***lse(contintion,data) 迭代過濾條件為false的資料。如果條件為空，返回data中為false的項；

booleans = [1, 0, 1, 0, 0, 1]


numbers = [23, 20, 44, 32, 7, 12]


print(list(itertools.filte***lse(none,booleans)))


#     [0, 0, 0]


print(list(itertools.filte***lse(lambda x : x < 20,numbers)))
# [23, 20, 44, 32]

返回我們需要使用的元素，根據b集合中元素真值，返回a集中對應的元素。

print(list(itertools.compress(letters,booleans)))

# [『a』, 『c』, 『f』]

​ 針對list中的每一項，呼叫函式功能。starmap(func,list) ；

starmap(pow, [(2,5), (3,2), (10,3)]) --> 32 9 1000


>>> from itertools import *


>>> x = starmap(max,[[5,14,5],[2,34,6],[3,5,2]])


>>> for i in x:


>>> print (i)


1434


5
repeat(object[, times]) 重複times次；

repeat(10, 3) --> 10 10 10
dropwhile(func, seq );當函式f執行返回假時, 開始迭代序列

dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
takewhile(predicate, iterable)；返回序列，當predicate為true是截止。

takewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
islice(seq[, start], stop[, step]);返回序列seq的從start開始到stop結束的步長為step的元素的迭代器

for i in islice("abcdef", 0, 4, 2):#a, c


print i
product(iter1,iter2, … itern, [repeat=1]);建立乙個迭代器，生成表示item1，item2等中的專案的笛卡爾積的元組，repeat是乙個關鍵字引數，指定重複生成序列的次數

# product('abcd', 'xy') --> ax ay bx by cx cy dx dy


# product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
for i in product([1, 2, 3], [4, 5], [6, 7]):


print i


(1, 4, 6)


(1, 4, 7)


(1, 5, 6)


(1, 5, 7)


(2, 4, 6)


(2, 4, 7)


(2, 5, 6)


(2, 5, 7)


(3, 4, 6)


(3, 4, 7)


(3, 5, 6)


(3, 5, 7)
permutations(p[,r]);返回p中任意取r個元素做排列的元組的迭代器

for i in permutations([1, 2, 3], 3):


print i


(1, 2, 3)


(1, 3, 2)


(2, 1, 3)


(2, 3, 1)


(3, 1, 2)


(3, 2, 1)
combinations(iterable,r);建立乙個迭代器，返回iterable中所有長度為r的子串行，返回的子串行中的項按輸入iterable中的順序排序

note:不帶重複

for i in combinations([1, 2, 3], 2):


print i


(1, 2)


(1, 3)


(2, 3)
同上, 帶重複 例子:

for i in combinations_with_replacement([1, 2, 3], 2):


print i


(1, 1)


(1, 2)


(1, 3)


(2, 2)


(2, 3)


(3, 3)
求質數序列中1,3,5,7,9,11,13,15三個數之和為35的三個數；

def get_three_data(data_list,amount):


for data in list(itertools.combinations(data_list, 3)):


if sum(data) == amount:


print(data)


#(7, 13, 15)


#(9, 11, 15)

python itertools 模組講解

1 介紹 itertools 是python的迭代器模組，itertools提供的工具相當高效且節省記憶體。使用這些工具，你將能夠建立自己定製的迭代器用於高效率的迴圈。無限迭代器 itertools包自帶了三個可以無限迭代的迭代器。這意味著，當你使用他們時，你要知道要的到底是最終會停止的迭代器，還是...

python itertools模組初學

usr bin python3 coding utf 8 time 2019 9 5 0005 20 18 author p.d site file demo.py import itertools 簡單來說就是累加。x itertools.accumulate range 101 print li...

Python itertools 操作迭代物件

python 的內建模組itertools提供了很多操作迭代物件的方法 返回乙個可無限迭代的迭代器，可以用於產生自然數 import itertools natuals itertools.count 1 1可以省略不屑，預設從0開始 for n in natuals print n 123 會無限...