python - 使用 numpy setdiff1d 保持顺序

Question

a = np.array([1, 2, 3])
b = np.array([4, 2, 3, 1, 0])
c = np.setdiff1d(b, a)
print("c", c)

结果是 c [0, 4] 但我想要的答案是 c [4 0]。

我该怎么做？

Answer 1

使用 np.in1d 获取不匹配的掩码，并简单地将 boolean-index 放入 b 以保留其中元素的顺序 -

b[~np.in1d(b,a)]

分步运行示例 -

In [14]: a
Out[14]: array([1, 2, 3])

In [15]: b
Out[15]: array([4, 2, 3, 1, 0])

In [16]: ~np.in1d(b,a)
Out[16]: array([ True, False, False, False,  True], dtype=bool)

In [17]: b[~np.in1d(b,a)]
Out[17]: array([4, 0])