Python 函数默认参数避坑指南

作者：和牛

2022 年 7 月 20 日
本文字数：1183 字
阅读完需：约 4 分钟

列表是一种经常使用的数据类型。在函数的定义中，常常会使用列表作为参数。

比如，要测试一个接口的数据，接口返回的数据格式如下：

{  "code": "20000",   "data": ["孙悟空","李白","甄姬"],   "msg": "success",   "status": 0}

复制代码

要测试的内容是：返回的 data 数据是否跟需求符合。在测试之前，需要对数据进一步处理，比如要增加 "王昭君" 这个元素进去，需要写一个函数：

def add_element(data=["孙悟空","李白","甄姬"]):    data.append('王昭君')    return data
print(add_element())print(add_element())print(add_element())

复制代码

在函数定义的时候经常会给参数设置默认值，在这个例子中，将 data 参数设置了默认值，函数定义以后，后面会被频繁的调用，期望值应该是打印如下：

["孙悟空","李白","甄姬","王昭君"]["孙悟空","李白","甄姬","王昭君"]["孙悟空","李白","甄姬","王昭君"]

复制代码

实际结果为：

["孙悟空","李白","甄姬","王昭君"]["孙悟空","李白","甄姬","王昭君","王昭君"]["孙悟空","李白","甄姬","王昭君","王昭君","王昭君"]

复制代码

原因

当定义函数时，会保存函数中默认参数 data 的值，也就是 ["孙悟空","李白","甄姬"],在每次调用的时候如果传递了新的实参，则使用传递的参数；没有传递，使用定义函数时保存的默认参数。

上面两次调用中，都没有传递新的实参，程序会调用定义函数时保存的默认参数，因为 append() , 在第一次调用以后，默认参数已经由 ["孙悟空","李白","甄姬"] 改变为 ["孙悟空","李白","甄姬","王昭君"],再次执行 append() 之后，就变成了 ["孙悟空","李白","甄姬","王昭君","王昭君"]；同理，第三次又改变了。

可以使用 id() 函数来定位问题：

def add_element(data=["孙悟空","李白","甄姬"]):    # id 来表示是不是同一个对象  print(id(data))    data.append('王昭君')    return data
print(add_element())print(add_element())print(add_element())

复制代码

打印出来的 id(data) 为同一个对象，也就是默认参数。如果我们改变第二个 print(add_element()) 为 print(add_element(["孙悟空","李白","甄姬"]))，那么第 2 个 id(data) 就会发生变化，因为它不在是默认值，而是新传进来的实参，实际结果也将变成：

2543416926792['孙悟空', '李白', '甄姬', '王昭君']2543418907848["孙悟空","李白","甄姬", '王昭君']2543416926792['孙悟空', '李白', '甄姬', '王昭君', '王昭君']

复制代码

改进方案

如果参数中有列表，尽量不要用它做默认参数
如果使用了列表作为默认参数，函数调用时传入实参，而不是省略
可以在函数体中另外定义一个变量接收默认参数

def add_element(data=["孙悟空","李白","甄姬"]):    if data == ["孙悟空","李白","甄姬"]:        data = ["孙悟空","李白","甄姬"]    data.append('王昭君')    return data

复制代码

我是九柄，公众号【九柄】，分享软件测试文章、面试、教程资料，欢迎来看看。

发布于: 刚刚阅读数: 3

和牛

关注

还未添加个人签名 2018.04.25 加入

python程序员，wx号：shoubian01

发布

暂无评论

创作场景

Python 函数默认参数避坑指南

原因

改进方案

和牛

评论