python - 用带有两个 header 信息的单个 header 替换多个数据帧 header

Question

我在 pandas 中使用了数据透视表并获得了所需的数据框格式，但现在我有两行标题。数据透视表后的结果数据框如下:

scenario     Actual               Plan
         LY_USD_AMT USD_AMT LY_USD_AMT USD_AMT
package
Africa            3       3          0       0
Brazil            1       1          1       1
Canada            1       1          1       1
Mexico            0       0          1       1

我已经成功地使用以下方法删除了标题的最后一行:

pd_piv.columns = pd_piv.columns.droplevel(-1)

但是在这一点上，很难识别哪一行是哪一行，因为它呈现的列名称如下

LY_USD_AMT     USD_AMT     LY_USD_AMT     USD_AMT

有没有办法解决这个问题，也许可以结合两个标题并获得一个更简单的表格数据框，如下所示。我需要一个简单的表格，因为我要把它提供给一个只能识别一个标题行的外部系统。

ACTUAL_LY_USD_AMT     ACTUAL_USD_AMT     Plan_LY_USD_AMT     Plan_USD_AMT

Answer 1

您可以组合两个 header :

df.columns = [c[0] + "_" + c[1] for c in df.columns]

这会将多个 header 更改为组合 header 。

例如:

我的数据框有多个标题:

    location                            location2
    S1          S2          S3          S1          S2          S3
a   -1.268587   0.014928    0.121195    -1.250765   0.321319    0.017481

上述代码的输出:

    location_S1 location_S2 location_S3 location2_S1    location2_S2    location2_S3
a   -1.268587   0.014928    0.121195    -1.250765   0.321319    0.017481