Заголовок столбца Python из сводной таблицы

#python #pandas #dataframe

#python #pandas #фрейм данных

Вопрос:

У меня есть приведенный ниже фрейм данных, который был сгенерирован из pivot_table :

                 cost    value
action           BUY     SELL
date                         
2001-04-27  79593.61      0.0
2001-05-04  29846.89  29132.6
2001-05-11  39786.30  40896.9
2001-05-18  29846.89  29550.5
2001-05-25  29844.71  29044.7
...              ...      ...
2020-08-28   9944.97   9955.3
2020-09-04   9944.04   9719.0
2020-09-11   9932.20  17751.5
2020-09-18  29849.57  29483.3
2020-09-25      0.00  86172.4
 

Однако теперь мне просто нужны три основных заголовка столбца date , cost , value . Но когда я пытаюсь присвоить столбцам имена через cost_df = cost_df.columns['cost','value'] , я получаю Index Error .

Ответ №1:

Ваша ось столбцов является многоиндексной. Если вам нужно выбрать данные, используя первый уровень мультииндекса, и удалить второй уровень, просто измените столбцы:

 cost_df.columns = ['cost', 'value']
cost_df[['cost', 'value']].head()

#                cost    value
# date                         
# 2001-04-27  79593.61      0.0
# 2001-05-04  29846.89  29132.6
# 2001-05-11  39786.30  40896.9
# 2001-05-18  29846.89  29550.5
# 2001-05-25  29844.71  29044.7