对于给定的月份，找出是否至少有一个8天的周期，其中7天标记为TRUE（python）

Question

好的，这是一个很难描述的问题

我正在为一项研究准备一份合规性报告，团队需要知道每个用户是否在8天中的7天内每月至少执行一次特定任务

因此，我需要一种方法：

1. 对于给定用户和给定月份，在该月份搜索8天期间，其中至少有7天标记为“真”
2. 如果满足要求，则返回1；如果不满足要求，则返回0

以下是数据结构的一个示例：

import pandas as pd

ids = 1
req_met = ['TRUE', 'TRUE', 'FALSE', 'FALSE', 'TRUE', 'TRUE', 'TRUE', 'FALSE', 'TRUE', 'TRUE', 'TRUE', 'TRUE', 'FALSE', 'FALSE', 'FALSE', 'FALSE', 'TRUE', 'TRUE', 'TRUE', 'FALSE']
date = ['2018-01-01', '2018-01-03', '2018-01-04', '2018-01-05', '2018-01-06', '2018-01-07', '2018-01-08', '2018-01-09', '2018-01-10', '2018-01-11',
'2018-01-12',  '2018-01-13', '2018-01-14', '2018-01-15', '2018-01-16', '2018-01-17', '2018-01-18', '2018-01-19', '2018-01-21', '2018-01-23']

df = pd.DataFrame({'id':ids,
                   'Date':date,
                   'req_met':req_met})
print(df)

    id        Date req_met
0    1  2018-01-01    TRUE
1    1  2018-01-03    TRUE
2    1  2018-01-04   FALSE
3    1  2018-01-05   FALSE
4    1  2018-01-06    TRUE
5    1  2018-01-07    TRUE
6    1  2018-01-08    TRUE
7    1  2018-01-09   FALSE
8    1  2018-01-10    TRUE
9    1  2018-01-11    TRUE
10   1  2018-01-12    TRUE
11   1  2018-01-13    TRUE
12   1  2018-01-14   FALSE
13   1  2018-01-15   FALSE
14   1  2018-01-16   FALSE
15   1  2018-01-17   FALSE
16   1  2018-01-18    TRUE
17   1  2018-01-19    TRUE
18   1  2018-01-21    TRUE
19   1  2018-01-23   FALSE

对于该用户，返回的答案为“1”，因为他们有8天的期限，其中7天为“真”（2018-01-06至2018-01-13）。您可以看到，日期范围并不总是连续的，这增加了复杂性

我期望的输出将是最有效的函数，它可以获取此数据并返回“1”（满足要求）或“0”（未满足要求）

提前感谢您的帮助

Answer 1

首先将类型转换为bool和datetime：

df['req_met'] = df['req_met'].replace({'TRUE':True, 'FALSE':False})
df['Date'] = pd.to_datetime(df.Date)

如果缺少日期，并且我们假设用户当天没有记录任何活动，我们需要插入缺少的日期：

df = (df.set_index('Date')
        .groupby('id').req_met
        .resample('D').sum()
        .reset_index()
        )

df['Month'] = df.Date.dt.strftime("%Y-%m")

请注意，resample用于每个id用户，以确保每个用户连续的日历日。 然后使用rolling方法：

df_result = (df.groupby(['id','Month'])
             .rolling(8)['req_met'].sum().ge(7)
             .groupby(['id','Month'])
             .agg({('req_met','max')})
             .reset_index()
             )

结果是：

   id    Month  req_met
0   1  2018-01     True

请注意groupby使用了两次。您可以通过逐步运行代码来检查计算，以完全理解逻辑