使用数据透视聚合

Question

我的数据框如下所示：

我想将其转换为以下形式：

我使用以下代码：

df2_SEX_AGE=df1.pivot(index='codprg', columns=['SEX','ETA'], values='counts')

但是会引发以下错误KeyError: 'Level SEX not found'

Answer 1

对于将MultiIndex转换为Index的情况，请使用列表理解和pivot_table以避免错误：

  

索引包含重复的条目，无法调整

---

df2_SEX_AGE=df1.pivot_table(index='codprg', 
                            columns=['SEX','ETA'], 
                            values='counts',  
                            aggfunc='sum')
df2_SEX_AGE.columns = [f'{a}{b}' for a, b in df2_SEX_AGE.columns]

Answer 2

这应该可以解决您的问题

(df1.assign(p=df['SEX'].astype(str) + df['ETA'].astype(str))
    .pivot(index='codprg', columns='p', values='counts'))