numpy.digitize(x, bins, right = False)
参数:
- x:一个数组
- bins:一个一维的数组,必须是升序或者降序
- right:间隔是否包含最右
- 返回值为x在bins中的位置
从我的理解来说,这个函数的含义可以用numpy.searchsorted()来理解,就是一个可以用来排序的函数。
【例】
import 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, right=True)
print(inds) # [1 2 3 4 4]
inds = np.digitize(x, bins, right=False)
print(inds) # [1 3 3 4 5]