函数

Last Updated: 2022-04-25

入门。

Python 函数的参数可以设默认值，比如这样一个简单的函数：

>>> def f(x, arr = []):
...     arr.append(x)
...     return arr
...

如果调用时不给arr，它将会使用默认的空数组

>>> f(2)
[2]

而如果给了arr则使用给定的数组：

>>> f(3, [1])
[1, 3]

放弃？

当多次调用函数时，神奇的事情发生了：

>>> def f(x, arr = []):
...     arr.append(x)
...     return arr
...
>>> f(1)
[1]
>>> f(2)
[1, 2]

比较一下 JavaScript 中的完全一样的函数：

> function f(x, arr=[]) {
... arr.push(x);
... return arr;
... }
undefined
> f(1)
[ 1 ]
> f(2)
[ 2 ]

进阶！

Python 的 default argument 是在函数定义的时候计算的，这不同于其他语言（比如 JavaScript 或 Ruby）在函数调用的时候计算，所以改变了 default argument 之后的调用也都会改变，这就是为什么最后一个调用f(4)的时候没有给arr赋值但arr却保留了之前的值[2,3]。

为了防止这种情况出现，可以用下面这个方法：

def f(x, arr=None):
  arr = arr or []
  arr.append(x)
  return arr