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