Align numpy array according to another array,"a[np.in1d(a, b)]"