Python高階部分?jǐn)U展 (collections ,itertools)
python3 collections擴(kuò)展
可命名元組 :namedtuple
from collections import namedtuple
stock_namedtuple = nametuple('stock',(date,txt))
stock_namedtuple('20190712','sun')
# 輸出
stock(date='20190712'.txt='sun')
有序字典
from collections import OrderedDict
deque
deque是為了高效實(shí)現(xiàn)插入和刪除操作的雙向列表,適合用于隊(duì)列和棧:
from collections import deque
q = deque(['a', 'b', 'c'])
q.append('x')
q.appendleft('y')
# 輸出
deque(['y', 'a', 'b', 'c', 'x'])
defaultdict
使用dict
時谅海,如果引用的Key不存在墩弯,就會拋出KeyError
。如果希望key不存在時彩库,返回一個默認(rèn)值威蕉,就可以使用
from collections import defaultdict
dd = defaultdict(lambda: 'N/A')
dd['key1'] = 'abc'
dd['key1'] # key1存在
# 輸出
'abc'
dd['key2'] # key2不存在玻粪,返回默認(rèn)值
# 輸出
'N/A'
高階函數(shù)
map()
函數(shù)接收兩個參數(shù),一個是函數(shù)项秉,一個是序列绣溜,map
將傳入的函數(shù)依次作用到序列的每個元素,并把結(jié)果作為新的list返回娄蔼。
def f(x):
return x * x
map(f, [1, 2, 3, 4, 5, 6, 7, 8, 9])
map(str, [1, 2, 3, 4, 5, 6, 7, 8, 9])
# 輸出
[1, 4, 9, 16, 25, 36, 49, 64, 81]
['1', '2', '3', '4', '5', '6', '7', '8', '9']
reduce() 把一個函數(shù)作用在一個序列[x1, x2, x3...]上怖喻,這個函數(shù)必須接收兩個參數(shù),reduce把結(jié)果繼續(xù)和序列的下一個元素做累積計(jì)算
def add(x, y):
return x + y
reduce(add, [1, 3, 5, 7, 9])
# 輸出
25
filter() 接收一個函數(shù)和一個序列岁诉。和map()
不同的時锚沸,filter()
把傳入的函數(shù)依次作用于每個元素,然后根據(jù)返回值是True
還是False
決定保留還是丟棄該元素涕癣。
def is_odd(n):
return n % 2 == 1
filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15])
# 輸出
[1, 5, 9, 15]
itertools的應(yīng)用
permutations(),考慮順序組合元素
>>> import itertools
>>> a = [1, 2, 3]
>>> b = itertools.permutations(a)
>>> b
# 輸出
(1, 2, 3), (1, 3, 2), (2, 1, 3), (2, 3, 1), (3, 1, 2), (3, 2, 1)
combinations() ,不考慮順序哗蜈,不放回?cái)?shù)據(jù)
import itertools
list1 = [1, 3, 4, 5]
list2 = list(itertools.combinations(list1, 2))
print(list2)
# 輸出
[(1, 3), (1, 4), (1, 5), (3, 4), (3, 5), (4, 5)]
st(itertools.combinations(list1, 2))
print(list2)
輸出
[(1, 3), (1, 4), (1, 5), (3, 4), (3, 5), (4, 5)]