Split Pivoted Index Column Pandas

时间:2017-03-21 03:36:57

标签: python pandas

我有一个如下所示的数据框:

                 |Units_sold | Revenue
 -------------------------------------
 California_2015 |     10    |    600   
 California_2016 |     15    |    900 

还有其他列,但基本上我想做的是将索引列拆散,并使我的表看起来像这样:

 |State     |Year  |Units_sold |Revenue
 -------------------------------------
 California |2015  | 10        |600   
 California |2016  | 15        |900 `

基本上我需要合并两个数据框,状态和年份,但我不确定如何拆分索引列/如果可能的话。对Python来说还是一个新手,所以我非常感谢任何输入!!

1 个答案:

答案 0 :(得分:1)

df = pd.DataFrame({'Units_sold':[10,15],'Revenue':[600,900]}, index=['California_2015','California_2016'])

df = df.reset_index()
df['State'] = df['index'].str.split("_").str.get(0)
df['Year'] = df['index'].str.split("_").str.get(1)
df = df.set_index('State')[['Year','Units_sold','Revenue']]
df