python - 在 python 中忽略 numpy bincount 中的 NaN

Question

我有一个一维数组，我想使用 numpy bincount 创建直方图。它工作正常，但我希望它忽略 NaN 值。

histogram = np.bincount(distancesArray, weights=intensitiesArray) / np.bincount(distancesArray)

我该怎么做？

感谢您的帮助!

Answer 1

我认为您的问题是:

import numpy

w = numpy.array([0.3, float("nan"), 0.2, 0.7, 1., -0.6]) # weights
x = numpy.array([0, 1, 1, 2, 2, 2])
numpy.bincount(x,  weights=w)
#>>> array([ 0.3,  nan,  1.1])

解决方案就是使用索引只保留非 nan 权重:

keep = ~numpy.isnan(w)
numpy.bincount(x[keep],  weights=w[keep])
#>>> array([ 0.3,  0.2,  1.1])