删除层次结构索引中的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：

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