jax.numpy.digitize

jax.numpy.digitize(x, bins, right=False)

返回输入数组中每个值所属的箱子的索引。

LAX-backend 实现的 numpy.digitize()。

原始文档字符串如下。

right	箱子的顺序	返回的索引 i 满足
False	增加	bins[i-1] <= x < bins[i]
True	增加	bins[i-1] < x <= bins[i]
False	减少	bins[i-1] > x >= bins[i]
True	减少	bins[i-1] >= x > bins[i]

如果 x 中的值超出 bins 的范围，则根据情况返回 0 或 len(bins)。

参数:

• x (array_like) – 要分箱的输入数组。在 NumPy 1.10.0 之前，这个数组必须是 1 维的，但现在可以有任意形状。

• bins (array_like) – 箱数组。它必须是单维且单调的。

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

返回:

indices – 输出索引数组，形状与 x 相同。

返回类型:

ndarray of ints