python高阶函数

200 阅读2分钟

「这是我参与2022首次更文挑战的第18天,活动详情查看:2022首次更文挑战

基本概念

在python中,可以接收参数是函数的函数,被称为高阶函数,函数式编程就是指这种高度抽象的编程范式,举个例子

def new_add(x, y, f):
    return f(x) + f(y)


def trans(a):
    if isinstance(a, str):
        return a
    if isinstance(a, int):
        return str(a)


c = new_add(1, "2", trans)
print(c)

python中常用的高阶函数

map()

map有两个参数,一个是函数,另一个参数是可迭代对象,map将传入的函数依次作用到序列的每个元素,并把结果作为新的Iterator返回。例如,给定一个整数列表,让列表中的每个元素都是原来的2倍,如果按照一般思路,我们会这样实现

int_list1 = [1, 2, 3, 4, 5]
int_list2 = []
for i in int_list1:
   int_list2.append(2*i)
print(int_list2)

那么用map我们可以这样写

f = lambda x: x * 2
list3 = map(f, int_list1)

所以,map()作为高阶函数,事实上它把运算规则抽象了,因此,我们不但可以计算简单的f(x)=2 * x,还可以计算任意复杂的函数,比如平方啊,转化为字符串啊等等。

reduce()

reduce把一个函数作用在一个序列[x1, x2, x3, ...]上,这个函数必须接收两个参数,reduce把结果继续和序列的下一个元素做累积计算,其效果就是序列求和,和python的sum函数很像 reduce不能直接使用,得从functiontools里导入

from functools import reduce
f = lambda x, y: x+y
d = reduce(f, int_list1)
print(d)

结果是15

dd = sum(int_list1)
print(dd)

image.png

filter()

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

例如,在一个list中,删掉偶数,只保留奇数,可以这么写:

def is_odd(n):
    return n % 2 == 1

list(filter(is_odd, [1, 2, 4, 5, 6, 9, 10, 15]))
# 结果: [1, 5, 9, 15]

sorted()

sorted函数可以对list进行排序:

sorted(list1)

sorted()函数也是一个高阶函数,它还可以接收一个key函数来实现自定义的排序,例如按绝对值大小排序:

sorted([36, 5, -12, 9, -21], key=abs)
# 结果[5, 9, -12, -21, 36]

如果想进行反向排序,还有一个参数是reverse=True

sorted(['bob', 'about', 'Zoo', 'Credit'], key=str.lower, reverse=True)
# 结果是:['Zoo', 'Credit', 'bob', 'about']