From efc1502a506d17f5122f1da17d7125e025fa1ecb Mon Sep 17 00:00:00 2001 From: wuaho Date: Fri, 11 Jun 2021 17:35:26 +0800 Subject: [PATCH] update --- api/api_v1/endpoints/query.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/api/api_v1/endpoints/query.py b/api/api_v1/endpoints/query.py index 4184bfb..8c8c224 100644 --- a/api/api_v1/endpoints/query.py +++ b/api/api_v1/endpoints/query.py @@ -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)