python - 带有浮点数的 Numpy bincount()

Question

我正在尝试获取浮点类型的 numpy 数组的 bincount：

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

如何将 bincount() 与浮点值而不是 int 一起使用？

Answer 1

numpy.unique使用前需要先使用bincount。否则，您所计算的内容是模棱两可的。unique对于 numpy 数组，应该比 Counter 快得多。

>>> w = np.array([0.1, 0.2, 0.1, 0.3, 0.5])
>>> uniqw, inverse = np.unique(w, return_inverse=True)
>>> uniqw
array([ 0.1,  0.2,  0.3,  0.5])
>>> np.bincount(inverse)
array([2, 1, 1, 1])

Answer 2

从1.9.0版本开始，可以np.unique直接使用：

w = np.array([0.1, 0.2, 0.1, 0.3, 0.5])
values, counts = np.unique(w, return_counts=True)

Answer 3

你想要这样的东西吗？

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

的输出对bincount浮点数没有意义：

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

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