puthonでデータフレームを使用した計算

処理No	数量	単価	金額	商品名
1	10	100	1000	A
1	5	50	250	B
2	-5	10	-50	B
1	20	10	200	C
2	-20	5	-100	C

処理No1=売上
処理No2=値引

B,Cは数量に対して値引き処理がされています。
商品名毎に販売された数量と合計金額を出したいです。
「groupby」を使用すると金額は良いのですが値引き分の数量も反映されてしまい
実際販売した数量が値引き分も計算されてしまい「0」になります。
何か良い方法はありませんでしょうか。
よろしくお願い致します。

欲しい結果
商品名　　数量　　金額
A 10 1000
B 5 200
C 20 100

import pandas as pd
df = pd.DataFrame (
  data={"処理No":[1,1,2,1,2],
        "数量":[10,5,-5,20,-20],
        "単価":[100,50,10,10,5],
        "金額":[1000,250,-50,200,-100],
        "商品名":["A","B","B","C","C"]})
df = pd.DataFrame(df,columns =["商品名","数量","金額"])
df2 = df.groupby("商品名").sum()

試した結果
商品名数量金額
A 　10 　1000
B 　0 　200
C 　0 　100

使っているツールのバージョンなど補足情報
Windows10
python3.9 64bit

行動規範の内容に同意します

回答1件

ベストアンサー

以下のようにしてください。

python
1>>> print(df)
2   処理No  数量  単価  金額 商品名
30       1    10   100  1000      A
41       1     5    50   250      B
52       2    -5    10   -50      B
63       1    20    10   200      C
74       2   -20     5  -100      C
8>>> df_modify = df.copy()
9>>> df_modify['数量'] = df_modify['数量'].mask(df_modify['処理No'] == 2, 0)
10>>> print(df_modify)
11   処理No  数量  単価  金額 商品名
120       1    10   100  1000      A
131       1     5    50   250      B
142       2     0    10   -50      B
153       1    20    10   200      C
164       2     0     5  -100      C
17>>> df_result = df_modify.groupby('商品名')[['数量', '金額']].sum()
18>>> print(df_result)
19        数量  金額
20商品名
21A         10  1000
22B          5   200
23C         20   100