python如何找到数组的元素在值数组内的位置

Question

我有一个很大的一维值数组，例如介于0和1之间的所有值。我有第二个数组，将[0,1]间隔划分为M个bin，我试图找到知道哪个bin的正确方法每个大数组值都适合。 像这样：

big = np.random.random(10000000)
bins = np.arange(0,1,0.01)
results = np.zeros_like(big)
for i,b in enumerate(big):
  results[i] = np.argmin(b>bins)

但是当然不需要昂贵的for循环。...

Answer 1

您要查找numpy.digitize（）吗？

https://docs.scipy.org/doc/numpy-1.15.1/reference/generated/numpy.digitize.html

numpy.digitize(big, bins)