在其他二维数组中查找二维数组

时间:2019-02-26 07:04:14

标签: python numpy

我确定这是重复的,但我找不到它。

我要检查较小的2d数组是否在较大的2d数组中,然后检查位置的索引或具有True / False的新数组

import numpy as np
arr = np.array([[1, 1, 1, 0, 0, 0, 0, 0],
                [0, 0, 0, 0, 0, 0, 0, 1],
                [1, 0, 1, 0, 1, 0, 1, 0],
                [0, 1, 0, 0, 1, 1, 0, 0],
                [0, 0, 1, 0, 1, 0, 0, 0],
                [0, 0, 1, 1, 1, 0, 1, 1],
                [1, 1, 1, 1, 0, 0, 0, 0],
                [1, 1, 1, 1, 1, 0, 0, 0]])

to_find = np.array([[1,1],
                    [1,1]])

enter image description here

这不起作用:

np.where(arr==to_find,1,0)
to_find in arr
to_find.tolist() in arr.tolist()

0 个答案:

没有答案