python - pandas 数据透视表，其中列包含具有多个类别的字符串

Question

我有以下形式的数据:

'cat'     'value'
a         1
a,b       2
a,b,c     3
b,c       2
b         1

我想使用数据透视表进行转换:

'a'  'b'  'c'
1
2    2
3    3    3
     2    2
     1

我如何执行此操作。如果我使用 pivot 命令:

df.pivot(columns= 'cat', values = 'value')

这产生了这个结果

'a' 'a,b' 'a,b,c' 'b,c' 'b' 
1
     2
           3
                   2
                        1

Answer 1

您可以使用 .explode()将字符串转换为列表后，然后正常旋转它:

df['cat'] = df['cat'].str.split(',')
df = df.explode('cat').pivot_table(index=df.explode('cat').index,columns='cat',values='value')

这输出:

cat a   b   c
0   1.0 NaN NaN
1   2.0 2.0 NaN
2   3.0 3.0 3.0
3   NaN 2.0 2.0
4   NaN 1.0 NaN

如果您不希望索引被命名为 cat，您可以重置或重命名索引。 .