我确定这是重复的,但我找不到它。
我要检查较小的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]])
这不起作用:
np.where(arr==to_find,1,0)
to_find in arr
to_find.tolist() in arr.tolist()