我正在编写一个辅助python函数,该函数将两个列表作为参数并对其进行迭代。通常,第二个列表必须是零数组,且长度与第一个相同。我想将第二个列表设置为关键字参数,并为其指定此默认值。
def iterate(list_a, list_b = np.zeros((len(list_a),), dtype=int)):
for a, b in zip(list_a, list_b):
# do something with a and b
但是,我收到一个未定义list_a的错误消息,这表明在定义函数时无法执行括号中的自引用计算。一个明显的解决方案是为list_b选择一个特殊的默认值,如果在调用该函数时保持不变,则使用if语句将其更改为零列表:
def iterate(list_a, list_b = 'zeros'):
if list_b == 'zeros':
list_b = np.zeros((len(list_a),), dtype=int)
for a, b in zip(list_a, list_b):
# do something with a and b
对于我来说,这种解决方案似乎并不十分精巧,我想知道是否有更好的方法来解决这个问题。
我将其保留为一般性目的,但是如果需要,我可以提供更多有关功能的详细信息。
答案 0 :(得分:1)
不,这无法完成,但是您已经以通常的方式解决了该问题。
通常,在这种情况下,人们会将默认值设置为None
并执行相同的操作(只是因为这种方式对于键入和输入变得不那么麻烦了-该函数接受数组或不接受任何内容,而不接受数组或字符串。 )
就可用性而言,您可以告诉用户doc脚本中默认值实际上是在做什么。
def iterate(list_a, list_b=None):
'list_b defaults to zero array of size a'
if not list_b:
list_b = np.zeros((len(list_a),), dtype=int)
for a, b in zip(list_a, list_b):
# do something with a and b
之所以不可能,是因为在同一行中调用list_a
变量时尚未创建,因为python解释器是逐行进行的。