Python標(biāo)準(zhǔn)庫itertools的使用方法
Python標(biāo)準(zhǔn)庫itertools模塊介紹
itertools是python內(nèi)置的模塊，使用簡單且功能強(qiáng)大，這里嘗試匯總整理下，并提供簡單應(yīng)用示例；如果還不能滿足你的要求，歡迎加入補(bǔ)充。
使用Python標(biāo)準(zhǔn)庫itertools只需簡單一句導(dǎo)入：import itertools
chain()
與其名稱意義一樣，給它一個(gè)列表如 lists/tuples/iterables，鏈接在一起；返回iterables對(duì)象。
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:])))
#輸出：{'a', 'd', 'b', 'e', 'c', 'f'}
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()
生成無界限序列，count(start=0, step=1) ，示例從100開始，步長為2，循環(huán)10，打印對(duì)應(yīng)值；必須手動(dòng)break，count()會(huì)一直循環(huán)。
i = 0
for item in itertools.count(100,2):
i += 1
if i > 10 : break
print(item)
filterfalse()
Python filterfalse(contintion,data) 迭代過濾條件為false的數(shù)據(jù)。如果條件為空，返回data中為false的項(xiàng)；
booleans = [1, 0, 1, 0, 0, 1] numbers = [23, 20, 44, 32, 7, 12] print(list(itertools.filterfalse(None,booleans))) #輸出：[0, 0, 0] print(list(itertools.filterfalse(lambda x : x < 20,numbers))) #輸出：[23, 20, 44, 32]
compress()
返回我們需要使用的元素，根據(jù)b集合中元素真值，返回a集中對(duì)應(yīng)的元素。
print(list(itertools.compress(letters,booleans))) # ['a', 'c', 'f']
starmap()
針對(duì)list中的每一項(xiàng)，調(diào)用函數(shù)功能。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) 14 34 5
repeat()
repeat(object[, times]) 重復(fù)times次；
repeat(10, 3) --> 10 10 10
dropwhile()
dropwhile(func, seq );當(dāng)函數(shù)f執(zhí)行返回假時(shí), 開始迭代序列
dropwhile(lambda x: x<5, [1,4,6,4,1]) --> 6 4 1
takewhile()
takewhile(predicate, iterable)；返回序列，當(dāng)predicate為true是截止。
takewhile(lambda x: x<5, [1,4,6,4,1]) --> 1 4
islice()
islice(seq[, start], stop[, step]);返回序列seq的從start開始到stop結(jié)束的步長為step的元素的迭代器
for i in islice("abcdef", 0, 4, 2):#a, c
print i
product()
product(iter1,iter2, ... iterN, [repeat=1]);創(chuàng)建一個(gè)迭代器，生成表示item1，item2等中的項(xiàng)目的笛卡爾積的元組，repeat是一個(gè)關(guān)鍵字參數(shù)，指定重復(fù)生成序列的次數(shù)
# 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()
permutations(p[,r]);返回p中任意取r個(gè)元素做排列的元組的迭代器
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()
combinations(iterable,r);創(chuàng)建一個(gè)迭代器，返回iterable中所有長度為r的子序列，返回的子序列中的項(xiàng)按輸入iterable中的順序排序
note:不帶重復(fù)
for i in combinations([1, 2, 3], 2): print i (1, 2) (1, 3) (2, 3)
combinations_with_replacement()
同上, 帶重復(fù) 例子:
for i in combinations_with_replacement([1, 2, 3], 2): print i (1, 1) (1, 2) (1, 3) (2, 2) (2, 3) (3, 3)
應(yīng)用示例
求質(zhì)數(shù)序列中1,3,5,7,9,11,13,15三個(gè)數(shù)之和為35的三個(gè)數(shù)；
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)
更多python標(biāo)準(zhǔn)庫使用方法請(qǐng)點(diǎn)擊下面的相關(guān)文章
相關(guān)文章
Django認(rèn)證系統(tǒng)user對(duì)象實(shí)現(xiàn)過程解析
這篇文章主要介紹了Django認(rèn)證系統(tǒng)user對(duì)象實(shí)現(xiàn)過程解析,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
二叉樹查找算法，在開發(fā)實(shí)踐中，會(huì)經(jīng)常用到。按照慣例，對(duì)于這么一個(gè)常用的東西，Python一定會(huì)提供輪子的。是的，python就是這樣，一定會(huì)讓開發(fā)者省心，降低開發(fā)者的工作壓力。2014-07-07
這篇文章主要介紹了python多進(jìn)程操作實(shí)例,本文講解了multiprocessing類的相關(guān)方法,然后給出了一個(gè)綜合實(shí)例和運(yùn)行效果,需要的朋友可以參考下2014-11-11
Python實(shí)現(xiàn)PS圖像抽象畫風(fēng)效果的方法
這篇文章主要介紹了Python實(shí)現(xiàn)PS圖像抽象畫風(fēng)效果的方法,涉及Python基于skimage模塊進(jìn)行圖像處理的相關(guān)操作技巧,需要的朋友可以參考下2018-01-01
namedtuple類位域Collections模塊中,有了namedtuple后通過屬性訪問數(shù)據(jù)能夠讓我們的代碼更加的直觀更好維護(hù),下面就來簡單講解Python編程中namedtuple類的用法2016-06-06
關(guān)于Python正則表達(dá)式 findall函數(shù)問題詳解
pd.drop_duplicates刪除重復(fù)行的方法實(shí)現(xiàn)