1

2021-08-24 10:23:08 +08:00 · 2021-08-24 10:23:08 +08:00 · 13c1eeac94
commit 13c1eeac94
parent 5325b31466
1 changed files with 2 additions and 1 deletions
--- a/api/api_v1/endpoints/query.py
+++ b/api/api_v1/endpoints/query.py
@ -157,6 +157,7 @@ async def retention_model(
    df = await ckdb.query_dataframe(sql)
    concat_data = []
    df = pd.concat([df, pd.DataFrame(concat_data, columns=df.columns)])
+    df['date'] = df['date'].apply(lambda x: x.date())
    # 计算整体
    summary_df = df.groupby('date')[['val_a', 'val_b', 'amount_a']].sum()
    summary_values = {}