Python| Numpy np.digitize() 方法

在np.digitize()方法的帮助下，我们可以使用np.digitize() () 方法获取每个值所属的 bin 的索引。

Syntax : np.digitize(Array, Bin, Right)
Return : Return an array of indices of the bins.

示例 #1：
在这个例子中，我们可以看到，通过使用np.digitize()方法，我们可以使用该方法获取属于数组的每个值的 bin 的索引数组。

# import numpy
import numpy as np
  
a = np.array([1.2, 2.4, 3.6, 4.8])
bins = np.array([1.0, 1.3, 2.5, 4.0, 10.0])
  
# using np.digitize() method
gfg = np.digitize(a, bins)
  
print(gfg)

输出 ：

[1 2 3 4]

示例 #2：

# import numpy
import numpy as np
  
a = np.array([[10.2, 21.4, 3.6, 14.8], [1.0, 5.0, 10.0, 15.0]])
bins = np.array([1.0, 1.3, 2.5, 4.0, 10.0])
  
# using np.digitize() method
gfg = np.digitize(a, bins)
  
print(gfg)

输出 ：

[[5 5 3 5]
[1 4 5 5]]