팬더 다른 열의 값에 따라 열의 합계를 집계하는 방법

Question

두 번째 열의 값에 대해 열의 값을 groupby으로 합산하려고 시도하지만 한편으로 세 번째 열의 값도 고려하면 df은 다음과 같습니다.

id memo amount 
1 pos  1.0 
1 pos  2.0 
1 neg  3.0 
2 pos  4.0 
2 pos  5.0 
2 neg  6.0 
2 neg  7.0 

난 id 그룹화하고 싶은 amount 합계 있지만, 각 그룹은 memo 경우 pos가 양 및 음의 예에 대한 것이다 neggroupby1 일 때, -1.0 - 2.0 + 3.0 = 0이므로 총량은 0입니다.

내가 df.groupby('id')['amount'].sum()을 수행하는 경우 과 amount 열만 고려하면 memo도 고려해야합니다.

그래서 결과가 모양을

,

id memo amount total_amount 
1 pos  1.0  0.0 
1 pos  2.0  0.0 
1 neg  3.0  0.0 
2 pos  4.0  -4.0 
2 pos  5.0  -4.0 
2 neg  6.0  -4.0 
2 neg  7.0  -4.0 

Answer 1

분할 두 단계로 작동, 당신은 당신이

df['temp'] = np.where(df.memo == 'pos', df.amount, -df.amount) 
df['total_amount'] = df.groupby('id').temp.transform(sum) 

Answer 2

을 통해 매핑 또 다른 재미있는 방법을 원하는 것을 얻을 수 있습니다 및

즉 곱

df['new'] = (df.set_index('id')['memo'].map({'pos':1,'neg':-1})*df['amount'].values)\ 
      .groupby(level=0).transform(sum).values 

출력 :

id memo amount new 
0 1 pos  1.0 0.0 
1 1 pos  2.0 0.0 
2 1 neg  3.0 0.0 
3 2 pos  4.0 -4.0 
4 2 pos  5.0 -4.0 
5 2 neg  6.0 -4.0 
6 2 neg  7.0 -4.0