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