Numpy: change max in each row to 1, all other numbers to 0

Question

I\'m trying to implement a numpy function that replaces the max in each row of a 2D array with 1, and all other numbers with zero:

>>> a = np.array(         


        

Answer 1

I prefer using numpy.where like so:

a[np.where(a==np.max(a))] = 1