Как включить пустой столбец в сводную таблицу Pandas - PullRequest
Новая
       26

Как включить пустой столбец в сводную таблицу Pandas

1 голос
/ 19 сентября 2019

У меня есть фрейм данных, в котором есть столбец «Комментарий» с пустыми значениями, и когда я использую сводную таблицу, столбец недоступен в выводе сводной таблицы

data_frame1 = pd.pivot_table(data_frame, index=['PC', 'Geo', 'Comp'], values=['Bill1', 'Bill2', 'Comment'], columns=['Month'], fill_value=0)
data_frame1 = data_frame1.swaplevel(0,1, axis=1).sort_index(axis=1)
tuples = [(a.strftime('%b-%y'), b) if a!= 'All' else (a,b) for a,b in data_frame1.columns]
data_frame1.columns = pd.MultiIndex.from_tuples(tuples)

Кадр входных данных 

PC Geo Comp  Month          Bill1 Bill2  Comment
A  Ind   OS  01/10/2019     1     1.28      
A  Ind   OS  01/11/2019     1     1.28   
A  Ind   OS  01/12/2019     1     1.28


выход 

               OCT-19             Nov-19        Dec-19
             Bill1 Bill2      Bill1 Bill2     Bill1 Bill2     
PC Geo Comp
A  Ind   OS   1     1.28        1    1.28      1    1.28


Желаемый выход 

                   OCT-19               Nov-19           Dec-19
             Bill1 Bill2 Comment Bill1 Bill2 Comment Bill1 Bill2 Comment     
PC Geo Comp
A  Ind   OS   1     1.28           1    1.28            1    1.28

1 Ответ

1 голос
/ 19 сентября 2019

Для меня рабочие изменения значений Comment после pivot_table, потому что если использовать до, столбец удаляется: 

data_frame1 = pd.pivot_table(data_frame, index=['PC', 'Geo', 'Comp'], 
                             values=['Bill1', 'Bill2', 'Comment'], 
                             columns=['Month'], 
                             fill_value=0)

data_frame1 = data_frame1.swaplevel(0,1, axis=1).sort_index(axis=1)

mask = data_frame1.columns.get_level_values(1) == 'Comment'
data_frame1.loc[:, mask] = ''
print (data_frame1)
Month       01/10/2019               01/11/2019               01/12/2019  \
                 Bill1 Bill2 Comment      Bill1 Bill2 Comment      Bill1   
PC Geo Comp                                                                
A  Ind OS            1  1.28                  1  1.28                  1   

Month                      
            Bill2 Comment  
PC Geo Comp                
A  Ind OS    1.28
...