cupy.searchsorted

cupy.searchsorted(a, v, side='left', sorter=None)

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

找到排序数组 a 中的索引，使得如果 v 中的对应元素插入到这些索引之前，a 的顺序将被保留。

参数:

• a (cupy.ndarray) – 输入数组。如果 sorter 是 None，那么它必须按升序排序，否则 sorter 必须是一个排序它的索引数组。

• v (cupy.ndarray) – 要插入到 a 中的值。

• side – {‘left’, ‘right’} 如果 left，返回找到的第一个合适位置的索引；如果 right，返回最后一个这样的索引。如果没有合适的索引，返回 0 或 a 的长度。

• sorter – 一维 array_like 可选数组，包含将数组 a 排序为升序的整数索引。它们通常是 argsort() 的结果。

返回:

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

返回类型:

cupy.ndarray

备注

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

参见

numpy.searchsorted()