在 DataFrame 中的新标题下嵌套列

Question

stocks= ['Apple','Raytheon','Amazon'] 
df = pd.DataFrame(data=[[10,11,12,13,5,2],[5,6,7,7,7,1]],
                        columns=['2020-12-31','2019-12-31','2020-09-26','2019-09-28','2020-01-01','2019-01-01'], index=['cash','inventory'])

我在数据框中有这个股票列表及其资产负债表。 我想将 Apple 下的前 2 列（2020-12-31,2019-12-31）、Raytheon 下的下两列和 Amazon 下的后两列分组。

如何在 Python 中实现这一点？

Answer 1

请尝试按比例的stocks列出了np.repeat再zip和使用MultiIndex.from_tuples ：

df.columns = pd.MultiIndex.from_tuples(zip(np.repeat(stocks, 2), df.columns))

df ：

               Apple              Raytheon                Amazon           
          2020-12-31 2019-12-31 2020-09-26 2019-09-28 2020-01-01 2019-01-01
cash              10         11         12         13          5          2
inventory          5          6          7          7          7          1