How to get the index of a maximum element in a NumPy array along one axis

python numpy max indices

>>> a.argmax(axis=0)array([1, 1, 0])

python numpy max indices

>>> import numpy as np>>> a = np.array([[1,2,3],[4,3,1]])>>> i,j = np.unravel_index(a.argmax(), a.shape)>>> a[i,j]4

python numpy max indices

If you ever need to do this for a shaped array, this works better than unravel:

import numpy as npa = np.array([[1,2,3], [4,3,1]])  # Can be of any shapeindices = np.where(a == a.max())

You can also change your conditions:

indices = np.where(a >= 1.5)

The above gives you results in the form that you asked for. Alternatively, you can convert to a list of x,y coordinates by:

x_y_coords =  zip(indices[0], indices[1])

CodeHunter