Python | Pandas TimedeltaIndex.searchsorted

在 Pandas 中，我们可以使用 TimedeltaIndex.searchsorted() 方法来搜索时差索引中的元素并返回插入这些元素以保持排序顺序所需的最小索引。该方法可以用于进行基于时间的插值以及其他时间序列分析。在本文中，我们将介绍 TimedeltaIndex.searchsorted() 方法的用法及示例。

语法

TimedeltaIndex.searchsorted(value, side='left')

参数说明：

• value（必选）：表示要在 TimedeltaIndex 中搜索的标量值或数组。
• side（可选）：表示在值的左侧还是右侧搜索。可以是“left”或“right”。默认值为“left”。

示例

首先，我们需要导入 pandas 模块。

import pandas as pd

然后，我们可以使用 pd.timedelta_range() 函数创建一个时间差范围。

tdi = pd.timedelta_range(start='1 day', periods=4, freq='12H')
print(tdi)

输出结果如下：

TimedeltaIndex(['1 days 00:00:00', '1 days 12:00:00', '2 days 00:00:00',
                '2 days 12:00:00'],
               dtype='timedelta64[ns]', freq='12H')

现在，我们可以使用 TimedeltaIndex.searchsorted() 方法来搜索 TimedeltaIndex 中的元素，并返回它应该插入的位置。

print(tdi.searchsorted(pd.Timedelta('30 hours')))
print(tdi.searchsorted(pd.Timedelta('34 hours'), side='right'))

输出结果如下：

2
3

在这个示例中，第一个搜索返回了 2，这意味着 Timedelta('30 hours') 应该插入索引位置 2 以保持有序。而第二个搜索返回了 3，这意味着 Timedelta('34 hours') 应该插入索引位置 3。注意，第二个搜索使用 side='right' 参数来搜索 TimedeltaIndex 的右侧。

结尾

本文介绍了 Pandas 中的 TimedeltaIndex.searchsorted() 方法，该方法用于搜索时差索引中的元素并返回插入这些元素以保持排序顺序所需的最小索引。本文提供了示例来解释 TimedeltaIndex.searchsorted() 的用法。