首页 > 解决方案 > 如何按功能将多列添加到数据框

问题描述

如果我有这样的df:

   a  b
0  1  3
1  2  4

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

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

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

标签: pythonpandasfunctiondataframemultiple-columns

解决方案


创建字典:

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

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

然后用你的df调用它:

df = new_columns(df, dictionary)

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

df.assign(**dictionary)

推荐阅读