Как сгруппировать элементы в кадре данных / списке, только если их суммы равны?

Question

Я хочу объединить элементы, если их купленные и проданные количества одинаковы на дату и оставлены без учета, как показано в примере ниже.

Набор данных:

 1/1/09 bought 100 AAPL
 1/1/09 sold 100 AAPL
 1/1/09 bought 35 AAPL
 1/1/09 sold 35 AAPL
 1/1/09 bought 105 AAPL

Я хочу, чтобы результирующий фрейм данных был:

1/1/09 bought 135 AAPL
1/1/09 sold 135 AAPL
1/1/09 bought 105 AAPL

Answer 1

Всего по группам и сумма:

my_var = ['bought','sold','bought','sold','bought']
dollar = [100,100,35,35,105]
date = ['2009-01-01','2009-01-01','2009-01-01','2009-01-01','2009-01-01']
sku = ['AAPL','AAPL','AAPL','AAPL','AAPL']

df = pd.DataFrame({'status':pd.Series(my_var),'dollars':pd.Series(dollar),'date':pd.Series(date),'SKU':pd.Series(sku)})

df.groupby(['SKU','status','date'],as_index = False)['dollars'].sum()