python - 从一个 numpy 数组中删除另一个 numpy 数组中的元素

Question

我想从一个 numpy 数组中减去另一个不存在的元素。

例子:

a = np.array([1,2,3,4])
b = np.array([1,2])

结果:

[3, 4]

Answer 1

您可以使用 Numpy 的 setdiff1d功能:

import numpy as np

a = np.array([1, 2, 3, 4])
b = np.array([1, 2])

c = np.setdiff1d(a, b)

print(c)

输出:

[3 4]

如果不想删除重复值，那么可以使用 Numpy 的 in1d功能:

import numpy as np

a = np.array([3, 1, 4, 2, 3, 4])
b = np.array([1, 2])

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

print(c)

输出:

[3 4 3 4]