比较两个 Numpy 数组并将不相等的向量保存在第三个数组中

Question

我有两个用向量填充的二维数组。 我想将 A 的所有向量与 B 中的所有向量进行比较，并保留 B 的所有向量不等于 A 中的向量。像这样在小范围内：

A = [[0,1,0], [1,1,0], [1,1,1]]

B = [[0,0,0], [1,0,0], [0,1,0], [1,1,0], [1,1,1]]

结果 = [[0,0,0], [1,0,0]]

我正在使用 numpy 并且似乎无法找到一种有效的方法来做到这一点。 我尝试使用两个 for 循环，这似乎非常低效，我尝试使用 for 循环，但这似乎非常低效：

for i in A: 
 for j in B: 
  if not np.all(A==B):
   print(B)

和 np.where，但这不会产生正确的结果。

for i in A:
 np.where(A!=1, A, False)

这对某些人来说可能很容易，但我非常感谢任何建议。

亲切的问候，

尼科

Answer 1

如果两个数组的大小都合适，则可以使用广播：

A = np.array([[0,1,0], [1,1,0], [1,1,1]])
B = np.array([[0,0,0], [1,0,0], [0,1,0], [1,1,0], [1,1,1]])

out = B[(A!=B[:,None]).any(2).all(1)]

输出：

array([[0, 0, 0],
       [1, 0, 0]])

或者，您可以使用 python 集：

a = set(map(tuple, A))
b = set(map(tuple, B))

out = np.array(list(b.difference(a)))

Answer 2

您可以简单地使用列表推导：

A = [[0,1,0], [1,1,0], [1,1,1]]

B = [[0,0,0], [1,0,0], [0,1,0], [1,1,0], [1,1,1]]

[x for x in B if x not in A]

#output
[[0, 0, 0], [1, 0, 0]]