如何在Numpy数组中找到矩形区域？

import numpy as np

image = np.array([[0, 0, 0, 0, 0],
                  [0, 1, 1, 1, 0],
                  [0, 1, 1, 1, 0],
                  [0, 0, 0, 0, 0]])

# 找到矩形区域的左上角和右下角坐标
rows, cols = np.where(image == 1)
top_left = (np.min(rows), np.min(cols))
bottom_right = (np.max(rows), np.max(cols))

import cv2

# 绘制矩形框
image_with_rectangle = cv2.rectangle(image, top_left, bottom_right, (0, 255, 0), 2)