Python | Numpy NP . digital()方法

原文:https://www . geesforgeks . org/python-numpy-NP-digital-method/

借助于**np.digitize()**方法,我们可以通过np.digitize()方法得到每个值所属的仓的索引。

语法: np.digitize(Array, Bin, Right) 返回:返回一组仓位的指数。

示例#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]]