cupy.searchsorted#

cupy.searchsorted(a, v, side='left', sorter=None)[source]#

找到应插入元素以保持顺序的索引。

找到已排序数组 a 中的索引，以便如果将 v 中的相应元素插入到这些索引之前，仍能保持 a 的顺序。

参数：

a (cupy.ndarray) – 输入数组。如果 sorter 为 None，则它必须按升序排序；否则，sorter 必须是用于排序该数组的索引数组。
v (cupy.ndarray) – 要插入到 a 中的值。
side – {'left', 'right'} 如果为 left，则返回找到的第一个合适位置的索引。如果为 right，则返回最后一个此类索引。如果没有合适的索引，则返回 0 或 a 的长度。
sorter – 1-D array_like 可选的整数索引数组，用于将数组 a 按升序排序。它们通常是 argsort() 的结果。

返回：

与 v 形状相同的插入点数组。

返回类型：

cupy.ndarray

注意

当 a 未按升序排序时，行为未定义。

另请参阅