对pandas数据帧中的某些行组求和

Question

我有一个pandas dataframe，我想在其中求和某些不规则间隔的日期之间的值。举个例子，我有这样的东西：

并且我知道这些总和的截止日期是2021-03-01、2021-03-04、2021-03-05，所以我想要截止日期和直到（但不包括）下一个截止日期产生的所有小部件的总和。所有不是截止日期的日期的值都为0。

这将产生如下所示的新列。

我怎么能在熊猫内部做到这一点呢？

Answer 1

您可以尝试使用：

df['date'] = pd.to_datetime(df['date']) #convert to datetime

然后检查是否出现截止值和累积值，并将其用作石斑鱼柱：

cutoffs= ['2021-03-01', '2021-03-04', '2021-03-05']
cond = df['date'].isin(pd.to_datetime(cutoffs))
df['sums'] = df.groupby(cond.cumsum())['widgets'].transform('sum').where(cond,0)

print(df)

        date  widgets    sums
0 2021-03-01        1       2
1 2021-03-02        0       0
2 2021-03-03        1       0
3 2021-03-04        3       3
4 2021-03-05        1       3
5 2021-03-06        2       0