Numpy bincount() with floats

Question

I am trying to get a bincount of a numpy array which is of the float type:

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

How can you use bincount() with float values and not int?

Answer 1

You need to use numpy.unique before you use bincount. Otherwise it's ambiguous what you're counting. unique should be much faster than Counter for numpy arrays.

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

Since version 1.9.0, you can use np.unique directly:

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