1. 程式人生 > 其它 >詳解python中三種高階函式(map,reduce,filter)

詳解python中三種高階函式(map,reduce,filter)

  • map(function,seq[,seq2]) 接收至少兩個引數,基本作用為將傳入的函式依次作用到序列的每個元素,並且把結果作為新的序列 返回一個可迭代的map物件

    • function:函式物件

      • py2中可為None,作用等同於zip()
        如:

      • py3中不可為None,None是不可呼叫、不可迭代物件

    • seq:可迭代物件,可以傳一個或多個

# 傳一個:
def func(i):return i*2
print([i for i in map(func,[1,'2'])]) # [2,'22']

# 傳多個
def func2(x,y):return x+y
print([i for i in map(func2,[1,2],[2,3])]) # [3, 5]
	圖解:


結合圖 map()的作用可以理解為:

# 傳一個時
seq=[1,'2']
result=[]
def func(x):return x*2
for i in seq:
	result.append(func(i))
print(result)
# 傳多個時
seq1=[1,2]
seq2=[2,3]
result=[]
def func2(x,y):return x+y
for x,y in zip(seq1,seq2):
	result.append(func2(x,y))
print(result)

當多個可迭代物件的長度不一致時,map只會取最短組合;同時每個可迭代物件相應位置引數型別需一致!(除了py支援的"str"*n)
如:

seq1=[1,2]
seq2=[2,3,4]
result=[]
def func2(x,y):return x+y
for x,y in zip(seq1,seq2):
	result.append(func2(x,y))
print(result) #[3,5]

map的function引數可以是lambda物件
如:

print([i for i in map(lambda x, y, z: (f'x:{x}', f'y:{y}', f'z:{z}'), [1, 2, 3, 4, 5], [1, 2, 3, 4], [1, 2])])
# [('x:1', 'y:1', 'z:1'), ('x:2', 'y:2', 'z:2')]
  • filter(function, seq)接收兩個引數,基本作用是對可迭代物件中的元素進行過濾;並返回一個新的可迭代filter物件

    • function:函式物件,返回值必須是個boolean值
    • seq:可迭代物件
      如:獲取所有小寫的字串
 	print([i for i in filter(lambda k: str(k).islower(), ['Java', 'Python', 'js', 'php'])])
 	# ['js', 'php']
等同於:
_list=['Java','Python','js','php']
result=[]
def is_lower(str_obj):return str(str_obj).islower()
for i in _list:
	if is_lower(i):
		result.append(i)
print(result)
  • reduce(function,seq[,initial])接收三個引數,基本作用為對序列進行累積;並返回結果。python3中reduce需從functools模組匯入

    • function:函式物件
    • seq: 可迭代物件
    • initial:初始值,選填引數
      工作過程是:
      reduce在迭代seq的過程中,第一次先把 seq的前兩個元素傳給 函式function,函式處理後,再把得到的結果和第三個元素作為兩個引數再次傳遞給函式function, 函式處理後得到的結果又和第四個元素作為兩個引數傳給函式function 依次類推,直至seq被迭代完。 如果傳入了 initial 值, 那麼首次傳遞的兩個元素則是 initial值 和 第一個元素。經過一次次累計計算之後得到一個彙總返回值。
      如:求和
def _add(x, y):
	return x + y
	
# 指定initial
print(reduce(_add,[1],3)) # 4
print(reduce(_add, [1, 2], 2)) # 5
# 不指定initial
print(reduce(_add, [1, 2])) # 3
print(reduce(_add,[1])) # 1
print(reduce(_add, [1, 2, 3, 4, 5])) # 15
等同於:
def fact(n):
	if n == 1:
		return 1
	return n + fact(n - 1)

print(fact(5)) # 15
藉助lambda:
print(reduce(lambda x, y: x + y, range(1, 6))) # 15

結合實際:假設我們要取出字典的key中包含某個關鍵字的鍵值對
如:取出下列字典中key值包含ECU的鍵值對

key = "ECU"
file_dict = {'value': 'name',
			 '刷寫ECU': 'burn_ecu_version=ecu_name,burn_package_url,(flash_method)',
			 'BD升級ECU': 'bd_ecu_version=ecu_name,doip_package_url',
			 '設定證書': 'set_ecu_certs=set_method,ecu_name,(bench_name)', 'x': {"ECU": "xx"}}

方法一:引入其他變數

result = {}
for k, v in file_dict.items():
	if key in k:
		result[k] = v
print(result) 
# {'刷寫ECU': 'burn_ecu_version=ecu_name,burn_package_url,(flash_method)', 'BD升級ECU': 'bd_ecu_version=ecu_name,doip_package_url'}

方法二:使用推導式

print(dict((k, v) for k, v in file_dict.items() if key in k))

方法三:reduce+map+filter

from functools import reduce
print(reduce(lambda x, y: x.update(y) or x,
			 [i for i in map(lambda k: {k: file_dict[k]}, filter(lambda k: key in k, file_dict))]))

細心的同學肯定發現無法過濾出巢狀key。這是弊端
解決方案:遞迴

class GetResource:
	def __init__(self):
		self.result = {}

	def get_resource(self, key_str, data):

		"""
		從dict中獲取包含指定key的k,v
		:param key_str:
		:param data:
		:return:
		"""

		if not isinstance(data, (dict, list, tuple)):
			pass
		elif isinstance(data, (list, tuple)):
			for index in data:
				self.get_resource(key_str, index)
		elif isinstance(data, dict):
			for k, v in data.items():
				if isinstance(v, str):
					if key_str in k:
						self.result[k] = v
				else:
					self.get_resource(key_str, v)
		return self.result


print(GetResource().get_resource(key, file_dict))
# {'刷寫ECU': 'burn_ecu_version=ecu_name,burn_package_url,(flash_method)', 'BD升級ECU': 'bd_ecu_version=ecu_name,doip_package_url', 'ECU': 'xx'}