python - Numpy bincount()与 float

我正在尝试获取numpy数组的bincount，该数组属于float类型：

w = np.array([0.1, 0.2, 0.1, 0.3, 0.5])
print np.bincount(w)

如何将bincount（）与float值而不是int一起使用？

最佳答案

你想要这样的东西吗？

>>> from collections import Counter
>>> w = np.array([0.1, 0.2, 0.1, 0.3, 0.5])
>>> c = Counter(w)

Counter({0.10000000000000001: 2, 0.5: 1, 0.29999999999999999: 1, 0.20000000000000001: 1})

或者，更漂亮的输出：

Counter({0.1: 2, 0.5: 1, 0.3: 1, 0.2: 1})

然后您可以对其进行排序并获取值：

>>> np.array([v for k,v in sorted(c.iteritems())])

array([2, 1, 1, 1])

对于float，bincount的输出没有意义：

>>> np.bincount([10,11])
array([0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 1, 1])

因为没有定义的浮动序列。