我正在尝试获取一个浮动类型的numpy数组的bincount:
w = np.array([0.1, 0.2, 0.1, 0.3, 0.5])
print np.bincount(w)
如何将bincount()与float值一起使用而不是int?
答案 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)