Как добавить вычисления в таблицу pandas groupby? - PullRequest
Новая
       125

Как добавить вычисления в таблицу pandas groupby?

1 голос
/ 18 июня 2020

Я создал таблицу, используя pandas .groupby, но хочу добавить столбец, который выполняет вычисления по столбцам в выводе groupby. Вот мой код (я использую Python 2) и результат: 

DMM_pivot = df.query('rpc_flag != "IB RPC"').groupby(['rpc_flag', 'DMM_type'])
DMM_pivot = DMM_pivot['number_accounts'].sum().unstack('rpc_flag').fillna(0)
DMM_pivot = DMM_pivot.apply(lambda x: x / x.sum()).round(2)  # Show percentage of total
print DMM_pivot

rpc_flag  No RPC  OB RPC
DMM_type                
Bad         0.34    0.25
Good        0.66    0.75

Я хочу добавить столбец, который показывает процентную разницу между столбцами: (OB RP C value / No RP C значение - 1) и ожидайте таких результатов: 

rpc_flag  No RPC  OB RPC   Change
DMM_type                
Bad         0.34    0.25    -0.26
Good        0.66    0.75     0.14

1 Ответ

0 голосов
/ 18 июня 2020

Применение применимо. 

DMM_pivot["Change"] = DMM_pivot.apply(lambda x: x["OB RPC"]/x["No RPC"] -1, axis=1)
...