python - 给定开始和结束日期的 Pandas，将它们分类为每周频率并计数

Question

我有开始数据和结束数据定义一个周期。我想将这些时间段汇总为每周，并计算每周分配的天数。

我想我必须将 pd.cut 与每周垃圾箱一起使用。我想首先将句点转换为日期列。然后使用 group by id 和 resample 来进行计数。我陷入了将周期转换为每日专栏的困境。

我很难通过关注Expanding pandas data frame with date range in columns来获得想要的结果

# dataframe
prod_id = ['a', 'a', 'b', 'c']
start_date = ['2019-01-15',  '2019-08-10', '2019-06-15',  '2019-07-10']
end_date = ['2019-01-28',  '2019-08-15', '2019-07-15',  '2019-07-22']
d = {'id': prod_id, 'start': start_date, 'end': end_date}
df = pd.DataFrame.from_dict(d)
df = df[['id', 'start', 'end']]
df['start'] = pd.to_datetime(df['start'])
df['end'] = pd.to_datetime(df['end'])

# bins
freq = 'W'
bins = pd.date_range(pd.to_datetime('2018-10-01'), pd.to_datetime('2019-10-20'), freq=freq).tolist()   

预期结果类似于

Answer 1

好吧，我想我找到了一种解决方法。这不是一个优雅的解决方案，因此仍在等待更好的解决方案。 (特别不喜欢每天重新采样)

正在关注Expanding pandas data frame with date range in columns

诀窍是引入另一列，这样groupby就可以单独区分重复的id。下面是代码

prod_id = ['a', 'a', 'b', 'c']
# add this to differetiate prod_id "a" 
num_id = [1, 2, 3, 4]
start_date = ['2019-01-15',  '2019-08-10', '2019-06-15',  '2019-07-10']
end_date = ['2019-01-28',  '2019-08-15', '2019-07-15',  '2019-07-22']
d = {'id': prod_id, 'nd': num_id, 'start': start_date, 'end': end_date}
df = pd.DataFrame.from_dict(d)
df = df[['id', 'nd', 'start', 'end']]
df['start'] = pd.to_datetime(df['start'])
df['end'] = pd.to_datetime(df['end'])    


df_melt = df.melt(id_vars=['id', 'nd'],value_name='date')
# first aggregate to daily
df_aggr_D = df_melt.groupby(['id', 'nd']).apply(lambda x: x.set_index('date').resample('D').pad()).drop(columns=['id','nd', 'variable']).reset_index()
df_aggr_D.drop(columns=['nd'], inplace=True)
# the aggregate to weekly
df_aggr_W = df_aggr_D.set_index('date').groupby('id').resample('W').count()
df_aggr_W.loc[df_aggr_W['id']!=0].rename(columns={'id':'count'})