Numpy bincount()与浮动

时间:2012-04-12 07:42:43

标签: python numpy

我正在尝试获取一个浮动类型的numpy数组的bincount:

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

如何将bincount()与float值一起使用而不是int?

3 个答案:

答案 0 :(得分:10)

在使用numpy.unique之前,您需要使用bincount。否则你的计算方式不明确。对于numpy数组,unique应该比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])

答案 1 :(得分:5)

你想要这样的东西吗?

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

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

答案 2 :(得分:4)

从版本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)