python - 我如何在 numpy 中复制这个 matlab 函数？

Question

[A,I] = histc([0.9828    0.4662    0.5245    0.9334    0.2163],[0.0191    0.2057    0.2820    0.2851    1.0000])

这是带有结果的 MATLAB 代码:

A =

     0     1     0     4     0


I =

     4     4     4     4     2

我需要的是我。我试过使用 np.histogram 但它给了我这个:

>>> a,b = np.histogram([0.9828 ,   0.4662 ,   0.5245 ,   0.9334 ,   0.2163],[0.0191   , 0.2057   , 0.2820  ,  0.2851  ,  1.0000])
>>> a
array([0, 1, 0, 4])
>>> b
array([ 0.0191,  0.2057,  0.282 ,  0.2851,  1.    ])

我想获取我的数组/矩阵中每个元素进入的容器。

Answer 1

你要找的是numpy.digitize:

Return the indices of the bins to which each value in input array belongs.

>>> a = np.digitize([0.9828 ,   0.4662 ,   0.5245 ,   0.9334 ,   0.2163],[0.0191   , 0.2057   , 0.2820  ,  0.2851  ,  1.0000])
>>> print(a)
[4 4 4 4 2]