我有一个数据框df
,如下所示:
Col1 Col2 Col3 StartDate EndDate Qty
24HR A1 B1 1/5/2020 2/28/2020 4.2
asd A2 B2 5/5/2020 7/15/2020 35
我想根据StartDate
和EndDate
之间的月份数重复数据帧的每一行,以获取以下内容:
Col1 Col2 Col3 StartDate EndDate Qty
24HR A1 B1 1/5/2020 1/31/2020 4.2
24HR A1 B1 2/1/2020 2/28/2020 4.2
asd A2 B2 5/5/2020 5/31/2020 35
asd A2 B2 6/1/2020 6/30/2020 35
asd A2 B2 7/1/2020 7/15/2020 35
我可以使用这种东西,但是它在逻辑上起作用:
df.StartDate = df.apply(lambda x: pd.date_range(start=x['StartDate'], end=x['EndDate'], freq='MS'), axis=1)
df = df.explode('StartDate')
我不确定如何解决这个问题。
答案 0 :(得分:2)
让我们进行快速修复
df['Date'] = df.apply(lambda x: pd.date_range(start=x['StartDate'], end=x['EndDate'], freq='D'), axis=1)
df=df.explode('Date')
df['Month']=df.Date.dt.strftime('%y-%m')
df=df.groupby(['Col1','Col2','Col3','Qty','Month']).Date.agg(['first','last']).reset_index()
df
Col1 Col2 Col3 Qty Month first last
0 24HR A1 B1 4.2 20-01 2020-01-05 2020-01-31
1 24HR A1 B1 4.2 20-02 2020-02-01 2020-02-28
2 asd A2 B2 35.0 20-05 2020-05-05 2020-05-31
3 asd A2 B2 35.0 20-06 2020-06-01 2020-06-30
4 asd A2 B2 35.0 20-07 2020-07-01 2020-07-15