Question

所以基本上我有两个数组，我想检查一个数组是否在另一个数组中...我正在寻找一种方法来做这样的事情：

>>> arr1 = np.array([1, 0, 0, 1, 1, 0])
>>> arr2 = np.array([0, 0, 1, 1, 1, 0])
>>> test_array = np.array([1, 1, 1])
>>> test_array in arr1
... False
>>> test_array in arr2
... True

有什么办法解决这样的事情吗？谢谢。

Answer 1

最直观的方法似乎是这样的迭代过程：

Start-Transcript -Path "C:\UserData.log" -Append

Answer 2

尝试使用2D蒙版：

ix = np.arange(len(arr1) - len(test_array))[:,None] + np.arange(len(test_array))
(arr1[ix] - test_array).all(axis=1).any()
>> False
(arr2[ix] - test_array).all(axis=1).any()
>> True

或在函数中：

def array_in(arr, test_arr):
    return (arr[np.arange(len(arr) - len(test_arr))[:,None] +
               np.arange(len(test_arr))] == test_arr).all(axis=1).any()

如何检查Numpy数组是否是另一个更大数组的子数组

2 个答案: