TypeError: searchsorted 需要兼容的 dtype 或标量，而不是 ndarray (Pandas)

Question

我无法在日期时间索引上应用 searchsorted 而不会出现错误：

df.index.searchsorted(pd.Series(np.array([df.index[0], df.index.max()])))

>>>TypeError: searchsorted requires compatible dtype or scalar, not ndarray

样本数据：

x = ['1998-01-07 13:17:00', '1998-01-07 13:18:00', '1998-01-07 13:19:00', '1998-01-07 13:20:00',
     '1998-01-07 13:21:00', '1998-01-07 13:22:00', '1998-01-07 13:23:00', '1998-01-07 13:24:00',
     '1998-01-07 13:25:00', '1998-01-07 13:26:00']
df = pd.DataFrame(x)
df.set_index(df.iloc[:, 0], inplace=True)
df.index = pd.DatetimeIndex(df.index.values)


df.index.searchsorted(pd.Series(np.array([df.index[0], df.index.max()])))

为什么它不起作用？ （听说是bug修复了，但不知何故它仍然存在）

我怎样才能让它工作？

我需要在多处理函数中应用这一行，它需要高效。

Answer 1

如错误所述，您将ndarray 传递给与df.index 系列不兼容的pandas.Series.searchsorted()，该系列的数据类型为pandas.DatetimeIndex。因此，要解决这个问题，您需要像这样调整给定系列的类型：

# just change this line
print(df.index.searchsorted(pd.DatetimeIndex([df.index[0], df.index.max()])))
# prints: [0, 9]