刪除層次結構索引中的Pandas頂級列

Question

我有一個像這樣的數據集

dataDate   prediction    original    
           Group pred
2018-09-30  A    2.30     3
2018-10-01  A    1.51     4
2018-10-02  A    2.08     5
2018-10-03  A    1.82     5
2018-09-30  B    0.96     2

我想刪除包含另外兩列的“預測”列，但是由於它是1級，所以我想做這樣的事情

df.drop('prediction')

但這給我一個錯誤

KeyError: "['prediction'] not found in axis"

從邏輯上講應該是可能的。

請幫助我通過上級列將其刪除，因為這將幫助我輕松刪除“預測”下的所有列

Answer 1

使用drop由第一級：

df = df.drop('prediction', axis=1, level=0)

然后通過MultiIndex.droplevel刪除第二級的MultiIndex.droplevel ：

df.columns = df.columns.droplevel(1)