Finding index of maximum value in array with NumPy,x[np.where(x == 5)]