Python Numpy np.digitize()用法及代码示例

借助np.digitize()方法，我们可以通过使用以下方法获取每个值所属数组的bin的索引：np.digitize()方法。

用法： np.digitize(Array, Bin, Right)

返回：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]]

相关用法

注：本文由纯净天空筛选整理自Jitender_1998大神的英文原创作品 Python | Numpy np.digitize() method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。