numpy: how to select rows based on a bunch of criteria,"a[np.in1d(a[:, (1)], b)]"