itertools常見方法的使用

2021-09-25 22:40:02 字數 2523 閱讀 9474

一、前言

本文介紹python標準庫itertools,為了方便,直接貼**實現。

二、**實現

1、排列

tmp = itertools.permutations([1, 2, 3], 2)

print(list(tmp))#[(1, 2), (1, 3), (2, 1), (2, 3), (3, 1), (3, 2)]

2、組合

tmp = itertools.combinations([1, 2, 3], 2)

print(list(tmp))#[(1, 2), (1, 3), (2, 3)]

3、flatten

tmp = itertools.chain.from_iterable([[1, 2], [3, 4]])

print(list(tmp))#[1, 2, 3, 4]

4、笛卡爾積

tmp = itertools.product([1, 2, 3], [4, 5])

print(list(tmp))#[(1, 4), (1, 5), (2, 4), (2, 5), (3, 4), (3, 5)]

5、建立ngram

import itertools

def n_gram(arr, n):

stop = none

tmp = (itertools.islice(arr, start, stop) for start in range(n))#第三個引數step預設為1

return zip(*tmp)

print(list(n_gram([1, 2, 3, 4, 5, 6], 2)))#[(1, 2), (2, 3), (3, 4), (4, 5), (5, 6)]

#上述過程的解釋

tmp1 = list(itertools.islice([1, 2, 3, 4, 5, 6], 0, none))#[1, 2, 3, 4, 5, 6]

tmp2 = list(itertools.islice([1, 2, 3, 4, 5, 6], 1, none))#[2, 3, 4, 5, 6]

t*** = (tmp1, tmp2)#([1, 2, 3, 4, 5, 6], [2, 3, 4, 5, 6])

tmp4 = zip(*t***)

print(list(tmp4))#[(1, 2), (2, 3), (3, 4), (4, 5), (5, 6)]

6、填充元組中的值

x = [1, 2, 3, 4, 5]

y = ['a', 'b', 'c']

res1 = zip(x, y)

print(list(res1))#[(1, 'a'), (2, 'b'), (3, 'c')]

res2 = itertools.zip_longest(x, y)

print(list(res2))#[(1, 'a'), (2, 'b'), (3, 'c'), (4, none), (5, none)]

7、累積計算(根據給出的函式,預設是求和)

tmp1 = itertools.accumulate([5, 15, 16, 2, 7])

print(list(tmp1))#[5, 20, 36, 38, 45]

tmp2 = itertools.accumulate([5, 15, 16, 2, 7], min)

print(list(tmp2))#[5, 5, 5, 2, 2]

t*** = itertools.accumulate([5, 15, 16, 2, 7], max)

print(list(t***))#[5, 15, 16, 16, 16]

8、返回謂詞為false的元素

tmp = itertools.filte***lse(bool, [none, false, 0, , 1])

print(list(tmp))#[none, false, 0, ]

9、從迭代中獲得引數來計算

import itertools

import operator

tmp = itertools.starmap(operator.mul, [(1, 2), (3, 4)])

print(list(tmp))#[2, 12]

10、返回謂詞為true元素,出現false則停止,所以如果判斷為x>4則不能得到結果

tmp1 = itertools.takewhile(lambda x : x < 4, [1, 2, 3, 4, 5, 6])

print(list(tmp1))#[1, 2, 3]

tmp2 = itertools.dropwhile(lambda x : x < 4, [1, 2, 3, 4, 5, 6])

print(list(tmp2))#[4, 5, 6]

Python os的常見方法

1 os.getcwd filename 相當於在當前執行檔案的目錄下建立乙個以filename命名的檔案 2 os.path.realpath file 獲取當前檔案路徑 c users admin pycharmprojects test case test unitest.py file 表示...

Java Collection常見方法

collection的常見方法 1 新增 boolean add object obj 新增單個物件 boolean addall collection coll 新增集合 2 刪除 boolean remove object obj 刪除單個物件 boolean removeall collect...

Java Set常見方法

set 元素不能重複,無序。set介面中的方法和collection一致 hashset 內部資料結構是雜湊表,是不同步的。雜湊表確定元素是否相同 1 判斷的是兩個元素的雜湊值是否相同,如果相同再判斷兩個物件的內容是否相同。2 判斷雜湊值相同,其實判斷的是物件的hashcode的方法,判斷內容相同,...