如何将数据透视头添加到数据帧中？

Question

我试图从数据库中收集一些数据，并使用pandas生成一个excel文件。 我得到的数据帧结构如下所示：

>>> df

                                            2020-05-21~2020-05-27
input_cost   jim    bob     companyc            17770.65                17770.65 
                            companye            3392.73                 3392.73
                            companyd            2892.30                 2892.30
                            companyf            22843.67                22843.67
             casy   dave    company_a           160.93                  160.93                                                      
                            company_b           160.93                  160.93

然后我想给它添加一个头，然后使用df.to_excel将它导出到excel文件

# expected output
                                                date
data_source  sale   leader  entity_name       2020-05-21~2020-05-27       sum 
input_cost   jim    bob     companyc            17770.65                17770.65
                            companye            3392.73                 3392.73
                            companyd            2892.30                 2892.30
                            companyf            22843.67                22843.67
             casy   dave    company_a           160.93                  160.93                                                                     
                            company_b           160.93                  160.93

对不起，我是新来的pandas，我怎样才能让它工作呢

谢谢

Answer 1

您可以使用rename_axis：

# rename index
df = df.rename_axis(['data_source', 'sale', 'leader', 'entity_name'])

# add level to columns
df.columns = pd.MultiIndex.from_product([['date'], df.columns])