Pandas DataFrame Count Elements по категориям - PullRequest
Новая
       11

Pandas DataFrame Count Elements по категориям

0 голосов
/ 13 апреля 2019

Рассмотрим следующий кадр данных Pandas, который показывает количество автомобилей, проданных разных марок и модельных лет: 

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», а столбец представляет собой комбинацию «год» и «количество продаж (многоуровневые столбцы хороши). Вывод должен быть примерно таким:это: 

           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 Ответ

1 голос
/ 13 апреля 2019

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
...