how to find the index of a value in 2d array in python?,np.where(a == 1)