展平熊猫数据透视表

时间:2016-09-01 13:54:06

标签: python pandas

这是我question的后续跟进。而不是数据透视表,是否可以将表格展平为如下所示:

data = {'year': ['2016', '2016', '2015', '2014', '2013'],
    'country':['uk', 'usa', 'fr','fr','uk'],
    'sales': [10, 21, 20, 10,12],
    'rep': ['john', 'john', 'claire', 'kyle','kyle']
    }

pd.DataFrame(data).pivot_table(index='country', columns='year', values=['rep','sales'])

          rep                       sales                  
year     2013  2014    2015  2016   2013  2014  2015  2016
country                                                  
fr       None  kyle  claire  None   None    10    20  None
uk       kyle  None    None  john    12  None  None    10
usa      None  None    None  john   None  None  None    21

展平表:

        rep_2013 rep_2014 rep_2015 rep_2016  sales_2013  sales_2014  sales_2015  sales_2016
country                                                  
fr       None    kyle     claire    None      None        10            20          None
uk       kyle    None     None      john      12          None          None         10
usa      None    None     None      john      None        None          None         21

2 个答案:

答案 0 :(得分:10)

请参阅collapse a pandas MultiIndex

解决方案

df.columns = df.columns.to_series().str.join('_')

答案 1 :(得分:2)

试试这个:

df.columns = df.columns.get_level_values(0)

接下来是:

df.columns = [' '.join(col).strip() for col in df.columns.values]

这应该会使你的多指数变平。