Pandas 在 Python 中查找日期范围重叠

在本文中，我们将介绍如何使用 Pandas 在 Python 中查找日期范围重叠的方法。日期范围重叠是指两个或多个时间段在某些时间段重叠的情况。

例如，假设你有两个时间段，一个是从 2021 年 1 月 1 日到 2021 年 1 月 10 日，另一个是从 2021 年 1 月 5 日到 2021 年 1 月 15 日。在这种情况下，日期范围从 2021 年 1 月 5 日到 2021 年 1 月 10 日重叠。

阅读更多：Pandas 教程

使用 Pandas 判断日期范围是否有重叠

首先，我们需要将日期范围转换为 Pandas 中的日期时间对象。要将字符串转换为 Pandas 日期时间对象，我们可以使用 Pandas 的 to_datetime() 函数。

import pandas as pd

df = pd.DataFrame({
    'start_date': ['2021-01-01', '2021-01-05', '2021-01-10'],
    'end_date': ['2021-01-10', '2021-01-15', '2021-01-20']
})

df['start_date'] = pd.to_datetime(df['start_date'])
df['end_date'] = pd.to_datetime(df['end_date'])
print(df)

输出结果：

  start_date   end_date
0 2021-01-01 2021-01-10
1 2021-01-05 2021-01-15
2 2021-01-10 2021-01-20

现在，我们可以使用 Pandas 的 intersect() 函数来检查是否有日期范围重叠。intersect() 函数返回两个日期范围的重叠部分。

overlap = df['start_date'].iloc[1:].intersect(df['end_date'].iloc[:-1])
print(overlap)

当你运行这段代码时，你应该会看到以下输出：

1   2021-01-10
dtype: datetime64[ns]

在这个例子中，我们可以看到 2021 年 1 月 10 日这一天是两个日期范围的重叠部分。

使用 Pandas 判断日期范围是否完全重叠

有时，你需要检查两个日期范围是否完全重叠。要检查两个日期范围是否完全重叠，你可以使用 Pandas 的 date_range() 函数与 timedelta() 函数。

df['duration'] = (df['end_date'] - df['start_date']).dt.days + 1

overlap = (pd.concat([pd.date_range(row.start_date, row.end_date, freq='d')
           for _, row in df.iterrows()])
    .value_counts()
    .sort_index()
    .eq(df['duration'])
    .all())

print(overlap)

输出结果：