如何在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.    ])

我想得到我的数组/矩阵中的每个元素都进入的bin.

Answer 1

你在寻找的是numpy.digitize:

返回输入数组中每个值所属的bin的索引.

>>> 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]