python - 检查一个 numpy 数组中的所有元素是否存在于第二个 numpy 数组中

Question

有没有人有一个更像numpy的方式来完成这个：

def uneven_compare(array1, array2):
    return numpy.all([numpy.any(array2 == elem) for elem in array1])

我只想检查一个数组中的所有元素是否存在于第二个数组中。

谢谢 ：）

Answer 1

使用numpy.in1d：

In [6]: array1 = np.array([0, 1, 2, 5, 0])

In [7]: array2 = np.array([0, 10, 20, 1, 2, 30, 5])

In [8]: np.in1d(array1, array2)
Out[8]: array([ True,  True,  True,  True,  True], dtype=bool)

In [9]: np.all(np.in1d(array1, array2))
Out[9]: True