python - 如何在函数内重新索引 pandas 数据框？

Question

我正在尝试将具有空值的列标题添加到我的数据帧( just like this answer )，但在已经修改它的函数内，如下所示:

mydf = pd.DataFrame()

def myfunc(df):
  df['newcol1'] = np.nan  # this works

  list_of_newcols = ['newcol2', 'newcol3']
  df = df.reindex(columns=df.columns.tolist() + list_of_newcols)  # this does not
  return
myfunc(mydf)

如果我在 IPython 控制台中单独运行这些行，它会添加它们。但作为脚本运行，newcol1 将被添加，但 2 和 3 不会。设置copy=False也不起作用。我在这里做错了什么？

Answer 1

Pandas df.reindex()除非索引相等，否则会生成一个新对象，因此您需要从函数中返回新对象。

def myfunc(df):
  df['newcol1'] = np.nan  # this works

  list_of_newcols = ['newcol2', 'newcol3']
  df = df.reindex(columns=df.columns.tolist + list_of_newcols)  # this does not
  return df

mydf = myfunc(mydf)