python - 如何将新列作为字符串分配给 Pandas 分配

Question

我如何将字符串作为 newcolumn 分配给 pandas.assign

例子:

newcolumn = 'myColumn'
    if(newcolumn != ''):
      sourcePandas = sourcePandas.assign(newcolumn = sourcePandas[row['PrimaryColumn']].apply(func))

以上不会引发任何错误，但会创建名称为 newcolumn 的新列相反 myColumn

Answer 1

您可以通过dict然后使用 ** 解压字典作为关键字参数传递给 pd.DataFrame.assign .

前任。:

newcolumn = 'myColumn'
df = pd.DataFrame()
df.assign(**{newcolumn:[0,1,2]})
   myColumn
0         0
1         1
2         2

所以它会

newcolumn = 'myColumn'
    if(newcolumn != ''):
      sourcePandas = sourcePandas.assign(**{newcolumn: sourcePandas[row['PrimaryColumn']].apply(func)})