每个类别的Pandas DataFrame计数元素

时间:2019-04-13 00:15:38

标签: pandas dataframe pivot-table

考虑以下熊猫数据框,该数据框显示了不同品牌和型号年份所售汽车的数量:

df = pd.DataFrame({'Y17': [ 1,       2,       1,     1,       0,       0,     1,       2,       2,     0],
                   'Y18': [ 0,       0,       1,     1,       1,       0,     0,       0,       0,     1],
                   'Make': ['Ford',  'Jeep',  'BMW', 'BMW',   'Mazda', 'Jeep', 'Chevy', 'BMW',  'Mazda','Ford']})

Make    Y17 Y18
Ford    1   0
Jeep    2   0
BMW     1   1
BMW     1   1
Mazda   0   1
Jeep    0   0
Chevy   1   0
BMW     2   0
Mazda   2   0
Ford    0   1

每一行都属于特定商店的销售额(未记录)。

我想旋转此数据框以获取条目数,以使索引为“ make”,列为“ year”和“ sales number”的组合(多级列可以)。输出应为像这样:

           Y17            Y18
        0   1   2      0   1   2
Make            
BMW     0   2   1      1   2   0
Chevy   0   1   0      1   0   0
Ford    1   1   0      1   1   0
Jeep    1   0   1      2   0   0
Mazda   1   0   1      1   1   0

1 个答案:

答案 0 :(得分:1)

melt然后crosstab

s=df.melt('Make')
pd.crosstab(s.Make,[s.variable,s.value]).stack().fillna(0).unstack()
variable Y17        Y18          
value      0  1  2    0    1    2
Make                             
BMW        0  2  1  1.0  2.0  0.0
Chevy      0  1  0  1.0  0.0  0.0
Ford       1  1  0  1.0  1.0  0.0
Jeep       1  0  1  2.0  0.0  0.0
Mazda      1  0  1  1.0  1.0  0.0