python之高阶函数和匿名函数

2023-01-30 23:01:30 函数 高阶 匿名

map()函数接收两个参数,一个是函数,一个是Iterable,map将传入的函数依次作用到序列的每个元素,并把结果作为新的Iterator返回。

1 def func(x):
2     return x*x
3 
4 r = map(func, [1, 2, 3, 4, 5])
5 print(type(r))
6 r = list(r)
7 print(r)

输出结果:

<class 'map'>
[1, 4, 9, 16, 25]

可以看出,map让函数func作用于列表的每一项,使列表的每一项都被函数func执行一次,即列表的每一项都进行平方。其返回值是map类型。

reduce函数必须接收两个参数,把一个函数作用在一个序列[x1, x2, x3, ...]上,然后再把结果继续和序列的下一个元素做累积计算。

 

1 from functools import reduce
2 
3 def fn(x, y):
4     return x * 10 + y
5 
6 f = reduce(fn, [1, 3, 5, 7, 9])   # 把序列变为整数
7 print(f)
8 print(type(f))

 

输出结果:

13579
<class 'int'>

和map不同,虽然reduce也是作用于每个元素,但是reduce的作用结果要用在下次和另一个元素做累积计算。

 1 from functools import reduce
 2 
 3 def fn(x, y):
 4     return x * 10 + y
 5 
 6 def char2num(s):
 7     digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
 8     return digits[s]
 9 
10 f = reduce(fn, map(char2num, '13579'))
11 print(f)

输出结果:

13579

可以将字符串类型转换为int类型

 

filter()函数用于过滤序列,接收一个函数和一个序列。和map()不同的是,filter()把传入的函数依次作用于每个元素,然后根据返回值是True还是False决定保留还是丢弃该元素。

1 def not_empty(s):
2     return s and s.strip()  # 不能直接写s.strip()
3 
4 f = filter(not_empty, ['A', '', 'B', None, 'C', '  '])
5 print(type(f))
6 print(list(f))

输出结果:

<class 'filter'>
['A', 'B', 'C']

sorted()函数也是一个高阶函数,在列表的学习中初步接触了sorted(),但其实它还可以接收一个key函数来实现自定义的排序

key指定的函数将作用于被排序对象的每一个元素上,并根据key函数返回的结果进行排序。

 1 l = sorted([36, 5, -12, 9, -21], key=abs)   # 按绝对值大小排序
 2 print(l)
 3 
 4 s = sorted(['bob', 'about', 'Zoo', 'Credit'])       # 按ASCII大小排序
 5 print(s)
 6 
 7 sl = sorted(['bob', 'about', 'Zoo', 'Credit'], key=str.lower)   # 忽略大小写排序
 8 print(sl)
 9 
10 sr = sorted(['bob', 'about', 'Zoo', 'Credit'], key=str.lower, reverse=True)     # 反向排序
11 print(sr)

输出结果:

[5, 9, -12, -21, 36]
['Credit', 'Zoo', 'about', 'bob']
['about', 'bob', 'Credit', 'Zoo']
['Zoo', 'Credit', 'bob', 'about']

 

同样的,sorted()也可以对元组和字典进行排序

1 from operator import itemgetter    # 需要使用operator模块
2 L = [('Bob', 75), ('Adam', 92), ('Bart', 66), ('Lisa', 88)]
3 print(sorted(L, key=itemgetter(0)))

输出结果:

[('Adam', 92), ('Bart', 66), ('Bob', 75), ('Lisa', 88)]

当然,也能以values的值来排序,对字典的排序与元组类似,但返回值不同。

1 from operator import itemgetter
2 
3 dic = {"Bob": 75, "Adam": 92, "Lisa": 88}
4 
5 print(sorted(dic,key=itemgetter(0)))
6 
7 print(sorted(dic, key=itemgetter(1)))

输出结果:

['Adam', 'Bob', 'Lisa']
['Adam', 'Lisa', 'Bob']

可以看到,不管是以key值进行排序,还是以value值进行排序,排序结果只返回对应顺序的key值。

1 f = list(map(lambda x: x * x, [1, 2, 3, 4, 5, 6, 7, 8, 9]))
2 print(f)

输出结果:

[1, 4, 9, 16, 25, 36, 49, 64, 81]

上面就是一个匿名函数的使用,匿名函数 lambda x: x * x 实际上就是:

1 def f(x):
2     return x * x

只是在这里没有显式地定义函数,这样因为函数没有名字,不必担心函数名冲突,而且代码看起来也简洁。

 

相关文章