阶乘的实现方式多种多样,可以递归、迭代、或者使用math库的阶乘 factorial 函数。但是这里我打算记录一个不同于这些比较简单、普遍的实现方式的,一个很有意思的新方法,使用reduce函数。
在这个方法中,我们需要用到 reduce 函数、int 类的__mul__方法和 range 函数。
reduce(int.__mul__, range(1, n+1))
如上代码,就是一个求 n 的阶乘 n! 的实现。这里 reduce 使用了python 内置库的 functools 库,所以需要提前引入库
from functools import reduce
对于 reduce 函数,进行一个简单的解读:
@overload
def reduce(function: Callable[[_T, _S], _T], sequence: Iterable[_S], initial: _T) -> _T: ...
@overload
def reduce(function: Callable[[_T, _T], _T], sequence: Iterable[_T]) -> _T: ...
从函数的定义可以看出来,reduce 接受两个或三个参数,其中参数分别是:
function:一个函数,注意这里是存储了函数的变量名,而不是带括号的函数调用
sequence:一个可迭代的对象
initial:初始值,可选
假设一个序列为 l = [ a1, a2, a3, a4, a5],同时有一个函数 func()
那么对于 reduce(func, l) 的解释是,先把 a1 和 a2 作为参数,调用 func 函数,然后将函数的返回值和 a3 作为参数,调用函数 func,以此类推,知道遍历完整个序列。写作
如果传入了参数 initial,则将 initial 的值和 a1 作为第一次的函数调用的参数。