This commit is contained in:
wuaho 2021-06-11 17:35:26 +08:00
parent 4aa6af0a85
commit efc1502a50

View File

@ -152,7 +152,7 @@ async def retention_model(
df.set_index(res['groupby'], inplace=True)
df.sort_index(inplace=True)
values = {}
days = [i for i in range((date_range[-1] - date_range[0]).days + 1)][:unit_num]
days = [i for i in range((date_range[-1] - date_range[0]).days + 1)][:unit_num+1]
for i, d1 in enumerate(date_range):
for g in groups:
a = set(df.loc[(d1, event_a, *g)]['values']) if (d1, event_a, *g) in df.index else set()
@ -171,7 +171,7 @@ async def retention_model(
'summary_values': summary_values,
'values': values,
'days': days,
'date_range': [d.strftime('%Y-%m-%d') for d in date_range][:unit_num],
'date_range': [d.strftime('%Y-%m-%d') for d in date_range][:unit_num+1],
'title': title
}
return schemas.Msg(code=0, msg='ok', data=data)