Расчет процентной разницы

Question

Я хочу знать, можно ли упростить следующий расчет или можно использовать альтернативные вычисления:

id      | group | value
01      |   1   |   5
01      |   1   |   6
02      |   1   |   35
02      |   1   |   40
03      |   1   |   90
03      |   1   |   95
control |   1   |   50
control |   1   |   60
04      |   2   |   35
04      |   2   |   36
05      |   2   |   15
05      |   2   |   10
06      |   2   |   20
06      |   2   |   25
control |   2   |   30
control |   2   |   40

Сначала вычисляется среднее значение всех элементов управления на группу:

id      | group | value_mean
control |   1   |   55
control |   2   |   35

Затем вычисляется средняя разница относительно контроля для каждого идентификатора:

id      | group | value_mean_percent
01      |   1   |   10
02      |   1   |   68
03      |   1   |   168
04      |   2   |   101
05      |   2   |   36
06      |   2   |   64

Пример для id = 1 (5 + 6) / 2 * 100/55 = 10%

Есть ли Другой способ интерпретировать результаты таких данных?

Answer 1

Я думаю, вы можете определить функцию, которая работает с подгруппой фрейма данных, где она группирует value по id, вычисляет среднее значение и возвращает все, разделенное на элемент управления:

def func(da):
    da = da.groupby('id')['value'].mean()
    return 100*da[da.index != "control"]/da['control']

Затем вы просто группируете data.frame по group, применяете эту функцию и сбрасываете индекс:

import pandas as pd
import numpy as np

df = pd.DataFrame({"id":np.repeat(['01','02','03','control','04','05','06','control'],2),
                  "group":np.repeat([1,2],8),
                   "value":[5,6,35,40,90,95,50,60,35,36,15,10,20,25,30,40]})

df.groupby('group').apply(func).reset_index()

    group   id  value
0   1   01  10.000000
1   1   02  68.181818
2   1   03  168.181818
3   2   04  101.428571
4   2   05  35.714286
5   2   06  64.285714