leguadmin/xbackend
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

update

Browse Source
This commit is contained in:
wuaho 2021-06-11 17:35:26 +08:00
parent 4aa6af0a85
commit efc1502a50
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
api/api_v1/endpoints/query.py
View File

@ -152,7 +152,7 @@ async def retention_model(
df.set_index(res['groupby'], inplace=True) df.set_index(res['groupby'], inplace=True)
df.sort_index(inplace=True) df.sort_index(inplace=True)
values = {} 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 i, d1 in enumerate(date_range):
for g in groups: for g in groups:
a = set(df.loc[(d1, event_a, *g)]['values']) if (d1, event_a, *g) in df.index else set() 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, 'summary_values': summary_values,
'values': values, 'values': values,
'days': days, '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 'title': title
} }
return schemas.Msg(code=0, msg='ok', data=data) return schemas.Msg(code=0, msg='ok', data=data)