В Pandas, как отправить результат преобразования groupby в исходный dataframe?

Question

Рассмотрим следующий пример

import pandas as pd 
import numpy as np 

df = pd.DataFrame({'A' : ['foo', 'bar', 'foo', 'bar', 
         'foo', 'bar', 'foo', 'foo'], 
        'B' : ['one', 'one', 'two', 'three', 
         'two', 'two', 'one', 'three'], 
        'D' : np.random.randn(8)}) 



group=df.groupby(['A','B']) 

agg_df=group.agg({'D':lambda x: x[x>0].sum(), 'D':lambda x: x[x<0].sum()}) 

Здесь я хотел бы получить две дополнительные переменные в исходной dataframe ДФ.

Один из них представляет собой сумму положительных элементов в D, а также сумму, которая представляет собой сумму отрицательных элементов в D. Использование agg является несгибаемым, как вы можете видеть в коде выше.

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

Наивный синтаксис будет использовать:

transform_df=group.transform({'D':lambda x: x[x>0].sum(), 'D':lambda x: x[x<0].sum()}) 

, но тот выходит из строя. Что я здесь делаю неправильно?

Благодаря

Answer 1

если они выражены в двух строках, логика становится чище писать & чтения

df['d_pos_sum'] = df.groupby(['A', 'B']).transform(lambda x: x[x>0].sum()) 
df['d_neg_sum'] = df.groupby(['A', 'B']).transform(lambda x: x[x<0].sum())