根据日期范围爆炸数据框

时间:2020-06-14 00:07:27

标签: pandas python-3.8

我有一个数据框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

我想根据StartDateEndDate之间的月份数重复数据帧的每一行,以获取以下内容:

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')

我不确定如何解决这个问题。

1 个答案:

答案 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