finding index of maximum value in array with numpy,np.where(x == 5)