如何通过函数向dataframe中添加多列

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)

Answer 2

def update_df(a_df, new_cols_names, new_cols_vals):
  for n, v in zip(new_cols_names, new_cols_vals):
    a_df[n] = v

update_df(df, ['c', 'd', 'e'], ['', 5, 6])