This commit is contained in:
wuaho 2021-06-11 16:43:44 +08:00
parent 9ddf741d3d
commit bfc2bd3a1a

View File

@ -141,7 +141,7 @@ async def retention_model(
continue
key = d1.strftime('%Y-%m-%d')
for j, d2 in enumerate(date_range[i:]):
if j >= unit_num:
if j > unit_num:
break
b = set(summary_df.loc[(d2, event_b)]['values']) if (d2, event_b) in summary_df.index else set()
tmp = summary_values.setdefault(key, {})
@ -160,7 +160,7 @@ async def retention_model(
continue
key = ','.join((d1.strftime("%Y-%m-%d"), *g))
for j, d2 in enumerate(date_range[i:]):
if j >= unit_num:
if j > unit_num:
break
b = set(df.loc[(d2, event_b, *g)]['values']) if (d2, event_b, *g) in df.index else set()
tmp = values.setdefault(key, {})