如何使用python获取矩阵中某个值的左上角和右下角坐标？

Question

我有一个2D numpy数组。让我们考虑下面的例子

[[0,6,89,255,1,6,7] 
    [0,255,89,255,1,1,7] 
    [255,255,255,255,255,255,255] 
    [1,2,3,4,5,6,7] 
    [1,1,255,5,7,3,255]] 

我们需要围绕某个值的框的坐标。例如，对于值255，围绕值255的框的坐标将是左上角（0,0）和右下角（4,6）。

如何在Python中有效地做到这一点。

非常感谢。

Answer 1

答案是非常相似：Is there a "bounding box" function (slice with non-zero values) for a ndarray in NumPy?

from numpy import array, argwhere 

A = array([[0 ,6 ,89 ,255,1 ,6 ,7 ], 
      [0 ,255,89 ,255,1 ,1 ,7 ], 
      [255,255,255,255,255,255,255], 
      [1 ,2 ,3 ,4 ,5 ,6 ,7 ], 
      [1 ,1 ,255,5 ,7 ,3 ,255]]) 

B = argwhere(A==255) 
(ystart, xstart), (ystop, xstop) = B.min(0), B.max(0)