добавьте столбцы, рассчитанные по группам, в фрейм данных, сохранив начальный порядок строк

#python #pandas #pandas-groupby #apply

Вопрос:

Python 3.8.5, Панды 1.1.4

 test_df = pd.DataFrame({
    'key': ['a', 'c', 'a', 'b', 'c', 'b', 'b', 'a', 'c', 'a'],
    'id': [859, 849, 238, 977, 427, 760, 453, 664, 102, 128],
    'order': [92, 32, 60, 4, 18, 19, 43, 69, 14, 88],
    'value': [18, 12, 16, 77, 62, 93, 86, 14, 49, 89]
})

test_df

    key id      order   value
0   a   859     92      18
1   c   849     32      12
2   a   238     60      16
3   b   977     4       77
4   c   427     18      62
5   b   760     19      93
6   b   453     43      86
7   a   664     69      14
8   c   102     14      49
9   a   128     88      89

 

Мне нужно рассчитать некоторые характеристики для каждой строки в группе ключей относительно столбца порядка, сохранить исходный порядок строк и добавить вычисленные характеристики в исходный фрейм данных.