Как проверить, является ли массив Numpy подмассивом другого большего массива

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

попробуйте использовать 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()

Answer 2

Наиболее интуитивным способом кажется итеративный процесс, подобный так:

def isSubset(arr1, arr2): 

    m = len(arr1)
    n = len(arr2)

    for i in range(0, n):
        for j in range(0, m):  
            if arr2[i] == arr1[j] 
                break; 
        """ 
        If the above inner loop was 
        not broken at all then arr2[i] 
        is not present in arr1 
        """
        if j == m:
           return False 
    """
    If we reach here then all 
    elements of arr2 are present 
    in arr1
    """
    return True