如何在 Matplotlib 中的图像上绘制矩形？

在使用 Matplotlib 绘制图像时，有时需要在图像上绘制矩形用于突出一些区域或标记一些位置。本文将介绍如何在 Matplotlib 中的图像上绘制矩形。

使用 Rectangle 方法

Matplotlib 中有一个专门用于绘制矩形的方法——Rectangle。此方法需要指定矩形的左下角坐标以及矩形的宽度和高度，如下所示：

import numpy as np
from matplotlib import pyplot as plt

# 创建一张图像
fig, ax = plt.subplots()

# 绘制一个矩形
rect = plt.Rectangle((0.2, 0.5), 0.4, 0.3, color='r', alpha=0.5)
ax.add_patch(rect)

# 显示图像
plt.show()

以上代码将在创建的图像上绘制了一个左下角为 (0.2, 0.5)、宽度为 0.4，高度为 0.3 的矩形。其中 color 参数指定矩形的颜色，alpha 参数指定矩形的透明度。

绘制多个矩形

有时需要在图像上绘制多个矩形，此时需要调用多次 Rectangle 方法。下面的代码演示了如何在图像上绘制两个矩形：

import numpy as np
from matplotlib import pyplot as plt

# 创建一张图像
fig, ax = plt.subplots()

# 绘制矩形1
rect1 = plt.Rectangle((0.2, 0.5), 0.4, 0.3, color='r', alpha=0.5)
ax.add_patch(rect1)

# 绘制矩形2
rect2 = plt.Rectangle((0.5, 0.2), 0.3, 0.4, color='b', alpha=0.5)
ax.add_patch(rect2)

# 显示图像
plt.show()

结语

本文介绍了在 Matplotlib 中绘制矩形的方法，并给出了代码示例。读者可以根据自己的需求进行相应的修改。