如何将数据透视表头添加到熊猫数据框中？

Question

我想从我的数据库中收集一些数据，并pandas用来制作一个 excel 文件。我得到的数据框结构如下：>>> df                                            2020-05-21~2020-05-27input_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然后我想给它添加一个标题，然后使用它导出到 excel 文件df.to_excel# expected output                                                datedata_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])