如何使用间隔索引连接具有重叠日期时间范围的数据帧？

使用区间索引连接具有重叠范围的数据帧

给定两个数据帧 df_1 和 df_2，具有表示日期时间范围的公共列，我们的目标是使用特定条件加入它们：df_1 的日期时间列值必须落在 df_2 中指定的范围内。

df_1

  timestamp              A          B
0 2016-05-14 10:54:33    0.020228   0.026572
1 2016-05-14 10:54:34    0.057780   0.175499
2 2016-05-14 10:54:35    0.098808   0.620986
3 2016-05-14 10:54:36    0.158789   1.014819
4 2016-05-14 10:54:39    0.038129   2.384590


df_2

  start                end                  event    
0 2016-05-14 10:54:31  2016-05-14 10:54:33  E1
1 2016-05-14 10:54:34  2016-05-14 10:54:37  E2
2 2016-05-14 10:54:38  2016-05-14 10:54:42  E3

解决方案：

我们可以使用区间索引来实现这一点。间隔索引根据 df_2 中指定的范围创建 bin，并将标签分配给 df_1 中属于这些 bin 的时间戳。

import pandas as pd

# Convert start and end columns to IntervalIndex
df_2.index = pd.IntervalIndex.from_arrays(df_2['start'], df_2['end'], closed='both')

# Get the event associated with each timestamp in df_1
df_1['event'] = df_1['timestamp'].apply(lambda x: df_2.iloc[df_2.index.get_loc(x)]['event'])

输出：

            timestamp         A         B event
0 2016-05-14 10:54:33  0.020228  0.026572    E1
1 2016-05-14 10:54:34  0.057780  0.175499    E2
2 2016-05-14 10:54:35  0.098808  0.620986    E2
3 2016-05-14 10:54:36  0.158789  1.014819    E2
4 2016-05-14 10:54:39  0.038129  2.384590    E3

以上是如何使用间隔索引连接具有重叠日期时间范围的数据帧？的详细内容。更多信息请关注PHP中文网其他相关文章！