mindspore.numpy.digitize

mindspore.numpy.digitize(x, bins, right=False)[源代码]

返回输入数组中每个值所属的桶的索引。如果 x 中的值超出了桶的界限，将返回 0 或 len(bins) 作为相应的索引。

参数：

  • x (Union[int, float, bool, list, tuple, Tensor]) - 待分桶的输入数组。

  • bins (Union[list, tuple, Tensor]) - 桶的数组。必须是一维且单调的。

  • right (boolean, 可选) - 表示区间是否包含右边界或左边界。默认为 (right==False) ，表示区间不包括右边界。在这种情况下，左边界是开放的，即 bins[i-1] <= x < bins[i] 是单调递增桶的默认行为。

返回：

元素为int的Tensor，输出元素为索引的数组，与 x 有相同的shape。

支持平台：

Ascend GPU CPU

样例：

>>> import mindspore.numpy as np
>>> x = np.array([1.2, 10.0, 12.4, 15.5, 20.])
>>> bins = np.array([0, 5, 10, 15, 20])
>>> inds = np.digitize(x, bins)
>>> print(inds)
[1 3 3 4 5]