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