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