python - 旋转并重命名 Pandas 数据框

Question

我有一个格式为的数据框

      Date         Datediff         Cumulative_sum
01 January 2019       1                   5
02 January 2019       1                   7
02 January 2019       2                   15
01 January 2019       2                   8
01 January 2019       3                   13

我想从数据框中旋转 Datediff 列，以便最终结果看起来像

Index            Day-1    Day-2    Day-3
01 January 2019    5        8        13
02 January 2019    7        15

我已经使用了pivot命令，因此

pt = pd.pivot_table(df, index = "Date",
                   columns = "Datediff",
                   values = "Cumulative_sum") \
               .reset_index() \
               .set_index("Date"))

返回数据透视表

                   1        2         3
01 January 2019    5        8        13
02 January 2019    7        15

然后我可以使用循环重命名列

for column in pt:
    pt.rename(columns = {column : "Day-" + str(column)}, inplace = True)

它返回的正是我想要的。但是，我想知道是否有一种更快的方法可以在旋转时重命名列并完全摆脱循环。

Answer 1

使用DataFrame.add_prefix :

df.add_prefix('Day-')

在您的解决方案中:

pt = (pd.pivot_table(df, index = "Date",
                   columns = "Datediff",
                   values = "Cumulative_sum")
        .add_prefix('Day-'))