从另一个 numpy 数组中删除一个 numpy 数组中的元素 [英] Remove elements in one numpy array from another numpy array

问题描述

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

I would like to subtract element from a numpy array that don't exist in another.

示例:

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

结果:

[3, 4]

推荐答案

你可以使用 Numpy 的 setdiff1d 函数:

You can use Numpy's setdiff1d function:

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 函数:

If duplicate values are not to be removed, then you can use Numpy's in1d function:

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]

这篇关于从另一个 numpy 数组中删除一个 numpy 数组中的元素的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！