有没有一种快速的方法可以检查一个日期是否在一个日期列表的n天内（比如7天）

Question

我正在使用以下数据集：

^{tb1}$

和一个列表holidays = ['2016-01-01','2016-01-18'....'2017-11-23','2017-12-25']

目标：创建一列，指示特定日期是否在列表中任何假日的+-7天内

模拟输出：

^{tb2}$

我正在处理大量的日期记录，因此试图找到一种快速（最优化）的方法来实现这一点

我当前的解决方案：

我想快速实现这一点的一种方法是创建另一个列表，其中只包含我所需持续时间（比如2年）的唯一日期。通过这种方式，我可以使用2 for loops实现一个简单的解决方案，以检查某个日期是否在假期的+-7天内，并且不会计算太多，因为两个列表都相对较小（730个唯一日期和假期列表中的约20个日期）。 一旦我有了我想要的日期列表，我所要做的就是在我的“日期”列上运行一次检查，看看该日期是否是我创建的新列表的一部分。但是，有什么建议可以更快地做到这一点吗

Answer 1

将假日转换为数据帧，然后^{}允许6天：

new_df = pd.merge_asof(df, holidays, left_on='Date', right_on='Holiday',
                       tolerance=pd.Timedelta(days=6))
new_df['Holiday'] = np.where(new_df['Holiday'].notnull(), 1, 0)
new_df = new_df.rename(columns={'Holiday': 'Within a week of Holiday'})

完整的工作示例：

import numpy as np
import pandas as pd

holidays = pd.DataFrame(pd.to_datetime(['2016-01-01', '2016-01-18']),
                        columns=['Holiday'])

df = pd.DataFrame({
    'Date': ['2016-01-04', '2016-01-05', '2016-01-06', '2016-01-07',
             '2016-01-08']
})
df['Date'] = pd.to_datetime(df['Date'])

new_df = pd.merge_asof(df, holidays, left_on='Date', right_on='Holiday',
                       tolerance=pd.Timedelta(days=6))
new_df['Holiday'] = np.where(new_df['Holiday'].notnull(), 1, 0)
new_df = new_df.rename(columns={'Holiday': 'Within a week of Holiday'})
print(new_df)

new_df：

        Date  Within a week of Holiday
0 2016-01-04                         1
1 2016-01-05                         1
2 2016-01-06                         1
3 2016-01-07                         1
4 2016-01-08                         0

---

或者将Holdiays转换为np datetime数组，然后对“Date”列进行^{}减法，将abs与7天进行比较，查看是否存在^{}匹配项：

holidays = np.array(['2016-01-01', '2016-01-18']).astype('datetime64')

df['Within a week of Holiday'] = (
        abs(df['Date'].values - holidays[:, None]) < pd.Timedelta(days=7)
).any(axis=0).astype(int)

完整的工作示例：

import numpy as np
import pandas as pd

holidays = np.array(['2016-01-01', '2016-01-18']).astype('datetime64')

df = pd.DataFrame({
    'Date': ['2016-01-04', '2016-01-05', '2016-01-06', '2016-01-07',
             '2016-01-08']
})
df['Date'] = pd.to_datetime(df['Date'])

df['Within a week of Holiday'] = (
        abs(df['Date'].values - holidays[:, None]) < pd.Timedelta(days=7)
).any(axis=0).astype(int)

print(df)

df：

        Date  Within a week of Holiday
0 2016-01-04                         1
1 2016-01-05                         1
2 2016-01-06                         1
3 2016-01-07                         1
4 2016-01-08                         0