python - Numpy直方图，如何在每个bin中取最大值

Question

我有一系列数字，我使用上面的代码进行分类。是否可以返回每个 bin 中的最大数量？

看一下示例代码：

  from numpy import *


  a=array([1,4,5,6,7.8,9,3.4,5.,6,3.5,6,8,9,10])

  bins=arange(0,11,1)

  h=hist(a,bins=bins)

  h=hist(a,bins=bins,weights=a)

这就是它的回报

  (array([  0. ,   1. ,   0. ,   6.9,   4. ,  10. ,  18. ,   7.8,   8. ,  28. ]), array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10])

我想知道是否有可能在第 4 个 bin 中获得 3.5（这是 3 到 4 之间的最大数字）而不是 6.9。

Answer 1

这将为您提供 bin 中每个元素的最大值，0如果 bin 中没有元素：

print [max(a[(a>=(i))&(a<i+1)]) if a[(a>=(i))&(a<i+1)].size else 0 for i in bins]
[0, 1.0, 0, 3.5, 4.0, 5.0, 6.0, 7.7999999999999998, 8.0, 9.0, 10.0]

更改+1您的垃圾箱大小，使其有用。

Answer 2

你可以使用numpy.digitize。请注意，它将小于第一个 bin 的值标记为 0。

a[np.digitize(a,bins) == 4].max()

掩码数组在这里很有用：

import numpy.ma as ma
a2 = ma.empty((len(bins),len(a)))
a2.data[...] = a
a2.mask = np.digitize(a,bins)-1 != bins[:,np.newaxis]
a2.max(axis=1).filled(np.nan)

array([  nan,   1. ,   nan,   3.5,   4. ,   5. ,   6. ,   7.8,   8. ,
         9. ,  10. ])