python - 如何通过函数向数据框添加多列

Question

如果我有这样的 df:

   a  b
0  1  3
1  2  4

我可以使用 df['c'] = '' 和 df['d'] = -1 添加 2 列并变成这样:

   a  b c  d
0  1  3   -1
1  2  4   -1

如何在函数中编写代码，以便我可以将该函数应用于 df 并一次添加所有列，而不是像上面那样一一单独添加？谢谢

Answer 1

创建字典:

dictionary= { 'c':'', 'd':-1 }

def new_columns(df, dictionary):
    return df.assign(**dictionary)

然后用你的 df 调用它:

df = new_columns(df, dictionary)

或者只是(如果您不需要函数调用，不确定您的用例是什么):

df.assign(**dictionary)