展平 pandas 数据透视表

Question

这是我的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

Answer 1

试试这个：

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

后跟：

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

这应该会使你的多索引变平

Answer 2

见collapse a pandas MultiIndex

解决方案

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