在给定范围内的2D NumPy数组比较

Question

如果我有一个二维数组数组，并且想要查看数组内的每个值是否在另一个范围内的另一个二维数组内，那么如何使用NumPy有效地进行操作？

[[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]] is in range 1 with 
[[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]] => TRUE 

[[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]] is in range 1 with 
[[0,3,0],[1,4,3],[1,4,5],[0,3,4],[0,4,3]] => TRUE 

[[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]] is in range 1 with 
[[0,4,0],[1,4,3],[1,4,5],[0,3,4],[0,4,3]] => FALSE 

最后一个是FALSE，因为关于指数0.1的项目的是4，这意味着ABS（2-4）> 1

Answer 1

可以与numpy的的矢量算术和all做到这一点很容易。例如：

>>> a = np.array([[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]]) 
>>> b = np.array([[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]]) 
>>> abs(a-b) 
array([[0, 0, 0], 
     [0, 0, 0], 
     [0, 0, 0], 
     [0, 0, 0], 
     [0, 0, 0]]) 
>>> abs(a-b) <= 1 
array([[ True, True, True], 
     [ True, True, True], 
     [ True, True, True], 
     [ True, True, True], 
     [ True, True, True]], dtype=bool) 
>>> (abs(a-b) <= 1).all() 
True 

和

>>> a2 = np.array([[1,2,1],[2,3,2],[2,3,4],[1,2,3],[1,3,2]]) 
>>> b2 = np.array([[0,4,0],[1,4,3],[1,4,5],[0,3,4],[0,4,3]]) 
>>> abs(a2-b2) <= 1 
array([[ True, False, True], 
     [ True, True, True], 
     [ True, True, True], 
     [ True, True, True], 
     [ True, True, True]], dtype=bool) 
>>> (abs(a2-b2) <= 1).all() 
False