Удалить 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 на MultiIndex.droplevel:

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