idx1 = np.all(np.abs(x) == 1, axis=0)