[Python进阶] 函数参数接收及判断(避坑)

2.20 函数参数接收及判断(避坑)

2.20.1 默认参数为可变类型

在Python中，我们经常会自定义函数，并且有些函数会带有参数。如果参数输入的是空列表，那么可能会如下进行定义：

def myfun(lst=[]):
    lst.append(1)
    return lst

这个函数看上去平平无奇，但是其实在使用起来会出现大问题。
我们连续调用3次看下效果：

ic(myfun())
ic(myfun())
ic(myfun())

15:32:46|> myfun(): [1]
15:32:47|> myfun(): [1, 1]
15:32:47|> myfun(): [1, 1, 1]

这是Python中的一个大坑，大家在定义默认参数时，如果遇到可变类型时，千万不要像这样定义默认值。
在Python中，默认参数只会求一次值，对于某些可变类型数据，指向的是同一个地址，于是发现会产生共用现象。
Python官方解释是：当函数被定义的时候，他的表达式只会被求值一次，然后每一次调用的时候，默认参数都会使用这个值。
改进代码：

from icecream import ic
def myfun(lst=None):
    if lst is None:
        lst = [1]
    return lst

ic(myfun())
ic(myfun())
ic(myfun())

15:43:56|> myfun(): [1]
15:43:56|> myfun(): [1]
15:43:56|> myfun(): [1]

2.20.2 默认参数判断

对于上面的问题，笔者在一些源码中看到有种默认参数判断并赋值的方式，如下：

from icecream import ic
def myfun(lst=None):
    l = lst or 5
    return l

ic(myfun())
ic(myfun())
ic(myfun())

15:58:53|> myfun(): 5
15:58:53|> myfun(): 5
15:58:53|> myfun(): 5

看上去好像是解决了，调用了3次，每次都成功的返回了正确的结果，但这种判断后赋值的方式也是可能存在问题的。比如：

from icecream import ic
def myfun(lst=None):
    l = lst or 5
    return l

ic(myfun(0))
ic(myfun(""))
ic(myfun(False))

16:00:57|> myfun(0): 5
16:00:57|> myfun(“”): 5
16:00:57|> myfun(False): 5

如果我们想传入0、````、False等Python中默认判断为False的值，则会由于会被判断为False，所以无法达到想要的效果。
应当改为：

from icecream import ic
def myfun(lst=None):
    l = 5 if lst is None else lst
    return l

ic(myfun(0))
ic(myfun(""))
ic(myfun(False))

16:04:01|> myfun(0): 0
16:04:01|> myfun(“”): ‘’
16:04:01|> myfun(False): False