Pandas фильтрует и суммирует, но применяется ко всем строкам ⇐ Python

[Anonymous]

У меня есть фрейм данных с идентификатором пользователя, кодом и значением.

Код: Выделить всё

user    code    value
0001       P       10
0001       P       20
0001       N       10
0002       N       40
0002       N       30
0003       P       10

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

Код: Выделить всё

user    code    value    Sum_of_P
0001       P       10          30
0001       P       20          30
0001       N       10          30
0002       N       40           0
0002       N       30           0
0003       P       10          10

Я пробовал сделать df['Sum_of_P'] = df.loc[df['code'] == 'P', 'value'].groupby(df['user'] ).transform('sum'), но это относится только к строкам с кодом = P. Есть ли способ применить это ко всем строкам?

Подробнее здесь: https://stackoverflow.com/questions/792 ... o-all-rows