高阶函数==>Higher-order function。
变量可以指向函数
以python内置的函数abs()为例,给一个参数-10,返回-10的绝对值10
>>>abs(-10)
10
如果只输入abs的话
>>> abs
<built-in function abs>
会有这样的返回,可见有参数的时候,即abs(-10)是函数调用,而abs是函数本身
要获得函数调用的结果,可以把结果赋值给一个变量
>>> x=abs(-10)
>>> x
10
但是这里如果把函数本身赋给变量
>>> f=abs
>>> f
<built-in function abs>
>>> f(-10)
10
可以发现函数本身可以赋给一个变量,即==>变量可以指向函数
如果一个变量指向了一个函数,就可以通过这个变量来调用函数,f(-10)实现了abs(-10)的功能。
函数名也是变量
函数名其实就是指向函数的变量。对于abs()这个函数,可以把它的函数名abs看作是一个变量,并且指向了一个可以计算绝对值的函数
因此你可以在交互环境中对abs赋值,比如abs=10,然后abs就不能算绝对值了,当然这么做并没有什么卵用,只是为了说明一下函数名也是变量
传入函数
变量可以指向函数,函数的参数能接收变量,那么一个函数就可以接收另一个函数作为参数,这种函数称为高阶函数
一个简单的高阶函数:
>>> def add(x,y,f):
return f(x)+f(y)
>>> add(3,4,abs)
7
add有三个参数,我们分别给3,4,abs
计算过程就是abs(3)+abs(4)==>7
关于map和reduce
map()函数接收两个参数,一个是函数,一个是Iterable,map将传入的函数依次作用于传入的序列的每个元素,并把结果作为新的Iterator返回
比如有一个函数f(x)=x^2,要把它作用在一个列表[1,2,3,4,5,6,7,8,9]上
>>> def f(x):
return x*x
>>> ls=map(f,[1,2,3,4,5,6,7,8,9])
>>> list(ls)
[1, 4, 9, 16, 25, 36, 49, 64, 81]
下面说reduce。reduce把一个函数作用在一个序列[x1,x2,x3...]上,这个函数必须接收两个参数,reduce把结果继续和序列的下一个元素做累积计算,效果就是:
reduce(f, [x1, x2, x3, x4]) = f(f(f(x1, x2), x3), x4)
比如对一个序列求和,通过reduce实现:
>>> from functools import reduce
>>> def add(x,y):
return x+y
>>> reduce(add,[1,3,5,7,9])
25
当然求和运算可以直接用Python内建函数sum()
,没必要动用reduce
。
但是如果要把序列[1, 3, 5, 7, 9]
变换成整数13579
,reduce
就可以派上用场:
>>> from functools import reduce
>>> def fn(x, y):
... return x * 10 + y
...
>>> reduce(fn, [1, 3, 5, 7, 9])
13579
这个例子本身没多大用处,但是,如果考虑到字符串str
也是一个序列,对上面的例子稍加改动,配合map()
,我们就可以写出把str
转换为int
的函数:
>>> from functools import reduce
>>> def fn(x, y):
... return x * 10 + y
...
>>> def char2num(s):
... digits = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
... return digits[s]
...
>>> reduce(fn, map(char2num, '13579'))
13579
整理成一个str2int
的函数就是:
from functools import reduce
DIGITS = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
def str2int(s):
def fn(x, y):
return x * 10 + y
def char2num(s):
return DIGITS[s]
return reduce(fn, map(char2num, s))
还可以用lambda函数进一步简化成:
from functools import reduce
DIGITS = {'0': 0, '1': 1, '2': 2, '3': 3, '4': 4, '5': 5, '6': 6, '7': 7, '8': 8, '9': 9}
def char2num(s):
return DIGITS[s]
def str2int(s):
return reduce(lambda x, y: x * 10 + y, map(char2num, s))
也就是说,假设Python没有提供int()
函数,你完全可以自己写一个把字符串转化为整数的函数,而且只需要几行代码!