如何按日期范圍匯總總和並加入

Question

我有一個客戶訂閱表 a，主要是 1 年合同。 期間從該月的第一天開始，到該月的最后一天結束。

,companyid,startdate,enddate
16216,10020659,2015-08-01,2016-07-31
23926,10020659,2016-08-01,2017-07-31
30078,10020659,2017-08-01,2018-07-31

有表 b 來跟蹤客戶的活動：

,companyid,isomonthofyear,count1,count2,count3
6325,10020659,2015-06-01,348,346,2
23605,10020659,2015-07-01,509,499,10
465310,10020659,2015-08-01,561,543,18
473875,10020659,2015-09-01,467,453,14
661421,10020659,2015-10-01,528,518,10
515050,10020659,2015-11-01,586,578,8
508636,10020659,2015-12-01,520,507,13
728064,10020659,2016-01-01,602,599,3
728024,10020659,2016-02-01,604,596,8
532500,10020659,2016-03-01,967,958,9
528642,10020659,2016-04-01,725,710,15
496834,10020659,2016-05-01,811,800,11
496701,10020659,2016-06-01,677,661,16
496682,10020659,2016-07-01,694,676,18
465301,10020659,2016-08-01,870,858,12

我只想匯總訂閱期的活動總和（count1-3），並加入表 a：

,companyid,startdate,enddate,sum_count1,sum_count2, sum_count3
 10020659.   2015-08-01,2016-07-31      10 20 30
 10020659,   2016-08-01,2017-07-31      14 589 29
,10020659,2017-08-01,2018-07-31         20 40 30

請注意，有非訂閱活動和不同的公司。 我想我們可以將結束日期轉換為每月的第一天。

Answer 1

首先確保類似日期的列是 dtype datetime：

df1['startdate']=pd.to_datetime(df1['startdate'])
df1['enddate']=pd.to_datetime(df1['enddate'])
df2['isomonthofyear']=pd.to_datetime(df2['isomonthofyear'])

最后：

out=df1.merge(df2,on='companyid',how='right')
#merging both df's on 'companyid' so to checking our criteria
m=out['isomonthofyear'].between(out['startdate'],out['enddate'])
#checking if it is under the subscription period or not
out=out[m].rename(columns={'counts':'count3'}).groupby('companyid').sum().add_prefix('sum_').reset_index()
#Filtering rows(selecting only those which are in subscription period) and then grouping them on companyid and performing sum
out=df1.merge(out,on='companyid',how='left')
#Finally merging the calculated sum with df1

out輸出：

    companyid   startdate   enddate     sum_count1  sum_count2  sum_count3
0   10020659    2015-08-01  2016-07-31     8612         8457    155
1   10020659    2016-08-01  2017-07-31     8612         8457    155
2   10020659    2017-08-01  2018-07-31     8612         8457    155