To get the index of the max value in a Numpy array, use np.argmax. Below is an example.
np.argmax
This can be helpful to get the highest probability in an array of probabilities.