带有None参数的Python自定义函数

Question

我想定义自己的函数如下：

def myown(df, ADD1, ADD2 = None, OtherArgument_1, OtherArgument_2):
tmp = df
tmp['NEWADD'] = (tmp['ADD1'] + ' ' + tmp['ADD2']).str.strip()
return tmp 

我知道这是不正确的，所以我可以在函数中添加if语句。

def myown(df, ADD1, ADD2 = None, OtherArgument_1, OtherArgument_2):
tmp = df
if ADD2 == None:
tmp['NEWADD'] = tmp[ADD1].str.strip()
else:
tmp['NEWADD'] = (tmp[ADD1] + ' ' + tmp[ADD2]).str.strip()

然而，如果我一开始不知道有多少ADD输入，我该如何修改
例如，这次需要组合5个ADD，下次可能是3个。每次都很难像这样重写函数：

def myown(df, ADD1, ADD2, ADD3, ADD4, ADD5, OtherArgument_1, OtherArgument_2):
tmp = df
tmp['NEWADD'] = (tmp[ADD1] + ' ' + tmp[ADD2] + ' ' + tmp[ADD3] + ' ' + tmp[ADD4] + ' ' + tmp[ADD5]).str.strip()

Answer 1

您可以通过使用以下循环和列表来实现这一点：

def myown(df, add_args, OtherArgument_1, OtherArgument_2):
tmp = df
new_add = ''
for i in add_args:
new_add = new_add + tmp[i].str.strip() + ''
tmp['NEWADD'] = new_add

add_args参数必须是一个列表，如下所示：

add_args = [ADD1, ADD2, ADDn]