python - 应用函数并返回多个数据帧后重命名系列列名称

Question

我有一个返回多个系列的函数，并且想要重命名系列名称。

 def myfunction(MyDataFrame, UniqueID, Var1, Var2):
     RunningSumA = MyDataFrame.groupby([UniqueID])[Var1].cumsum().reset_index()
     RunningSumB = MyDataFrame.groupby([UniqueID])[Var2].cumsum().reset_index()

     return pd.Series([RunningSumA, RunningSumB])


 Result = myfunction(MyDataFrame, UniqueID, Var1, Var2)

当我查看结果系列的前几行时，结果如下:

Result[0].head()

  index    0
0  216646  0
1  216647  0
2  216648  0
3  216649  0
4  216650  0

我想将 index 和 0 重命名为函数内有意义的名称。我尝试过重命名列，但无法重命名它们。我怎样才能做到这一点？

Answer 1

reset_index接受一个名称参数来填充列名称:

...
RunningSumA = MyDataFrame.groupby([UniqueID])[Var1].cumsum().reset_index(name="suma")
RunningSumB = MyDataFrame.groupby([UniqueID])[Var2].cumsum().reset_index(name="sumb")
...