Pandas 數據透視表多索引列到單級

Question

我有一個如下所示的熊貓數據框：

df = 
    COLUMN_NAME  YEAR1  YEAR2   VALUE
0   Column1       2013   2014   0.042835
1   Column1       2014   2015   0.033600
2   Column2       2013   2014   0.004406
3   Column2       2014   2015   0.016900
...

對於每個 COLUMN_NAME、YEAR1 和 YEAR2，都會計算一個 VALUE。 我想對數據框進行分組，使其在 COLUMN_NAME 上是唯一的，其中的列如下所示：

df_desired = 
    COLUMN_NAME  Value_from_2013_2014   Value_from_2014_2015 ...
0   Column1      0.042835                  0.033600
1   Column2      0.004406                  0.016900
...

我可以使用下面的代碼實現我想要的東西，但它創建了一個 MultiIndex 列，我該如何實現呢？ 謝謝您的幫助。

pd.pivot_table(df, 'VALUE', 'COLUMN_NAME', ['YEAR1', 'YEAR2'])

YEAR1         2013      2014
YEAR2         2014      2015
COLUMN_NAME     
Column1       0.042835  0.0336
Column2       0.004406  0.0169

Answer 1

您可以使用to_flat_index展平多to_flat_index列，然后map到str並添加您的前綴：

s.columns = ["Value_from_"+"_".join(map(str, i)) for i in s.columns.to_flat_index()]

print (s)

             Value_from_2013_2014  Value_from_2014_2015
COLUMN_NAME                                            
Column1                  0.042835                0.0336
Column2                  0.004406                0.0169

Answer 2

df_agg = pd.pivot_table(df, 'VALUE', 'COLUMN_NAME', ['YEAR1', 'YEAR2'])
df_agg.columns = ['Value_from_' + str(df_agg.columns[i][0]) + '_' + str(df_agg.columns[i][1]) for i in range(len(df_agg.columns))]

Answer 3

如果我理解正確，您將獲得所需的值，但不是所需的標簽。 如果是這樣，您可以使用df_desired.columns = ['Value_from_2013_2014', 'Value_from_2014_2015']更改列名