在第n行之后和之前在Pandas中求和

Question

我是Python新手并学习Pandas。我有一个下面给出的数据集，需要在购买第一个GPS产品之前将Amount列中的所有行加起来，并在购买第一个GPS产品到第二次GPS购买之后执行相同操作。

这是数据集：

df = pd.DataFrame({Product : [Sub,Sub,Sub,Sub,GPS,Sub,Sub,GPS,Sub,Sub]} {Amount :[13,15,25,32,43,76,23,45,67,89]})

提前感谢您的帮助。

Answer 1

我会使用iloc索引：

# locate first ocurence of purchase
first_GPS_index = df[df['Product']==GPS].index.min()
# use iloc to index same way as you would in python list
sum_until = df.iloc[:first_GPS_index].sum()
sum_after = df.iloc[first_GPS_index+1:].sum()