根据组(pandas dataframe)计算多列中的唯一值

时间:2018-06-06 01:18:37

标签: python pandas dataframe pivot-table

我在pandas中有一个数据框:

    a  b  c  d  e
0   1  x  y  t  u
1   1  x  z  y  v
2   1  y  x  t  z
3   2  x  t  z  u
4   2  z  x  y  t 
5   3  x  z  y  t

我需要将此数据帧转换为:

   a  x  y  z  t  u  v  y  z
0  1  3  3  2  2  1  1  3  2
1  2  2  1  2  2  1  0  1  2
2  3  1  1  1  1  0  0  0  0

2 个答案:

答案 0 :(得分:3)

IIUC

df.set_index('a').stack().groupby(level=0).value_counts().unstack(fill_value=0)
Out[514]: 
   t  u  v  x  y  z
a                  
1  2  1  1  3  3  2
2  2  1  0  2  1  2
3  1  0  0  1  1  1

或使用melt + crosstab

s=df.melt('a')
pd.crosstab(s.a,s.value)
Out[518]: 
value  t  u  v  x  y  z
a                      
1      2  1  1  3  3  2
2      2  1  0  2  1  2
3      1  0  0  1  1  1

答案 1 :(得分:1)

df.melt(id_vars="a").groupby(["a","value"]).count().unstack().fillna(0)

      variable                         
value        t    u    v    x    y    z
a                                      
1          2.0  1.0  1.0  3.0  3.0  2.0
2          2.0  1.0  0.0  2.0  1.0  2.0
3          1.0  0.0  0.0  1.0  1.0  1.0