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

均值留存

Browse Source
This commit is contained in:
wuaho 2021-11-30 11:33:17 +08:00
parent dbe3fb52b1
commit 47c62fd634
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -6,6 +6,7 @@ from urllib.parse import quote
import pandas as pd
import numpy as np
from fastapi import APIRouter, Depends, Request
from fastapi.encoders import jsonable_encoder
from motor.motor_asyncio import AsyncIOMotorDatabase
from fastapi.responses import StreamingResponse
@ -356,9 +357,13 @@ async def retention_model(request: Request,
tmp['d0'] = 0
for rt, rd in retention_avg_dict.items():
tmp['d0'] = tmp['d0'] + rd['cnt0']
tmp['p'].append(round(rd['cntn'] * 100 / rd['cnt0'], 2))
n = round(rd['cntn'] * 100 / rd['cnt0'],2)
n = 0 if np.isnan(n) else n
tmp['p'].append(n)
tmp['n'].append(rd['cntn'])
tmp['p_outflow'].append(round(rd['o_cntn'] * 100 / rd['cnt0'], 2))
n = round(rd['o_cntn'] * 100 / rd['cnt0'],2)
n = 0 if np.isnan(n) else n
tmp['p_outflow'].append(n)
tmp['n_outflow'].append(rd['o_cntn'])