Python在图像上绘制矩形

在Python的图像处理中，经常需要在图像上绘制矩形标志出某些区域。本文将介绍如何在Python中使用Pillow库绘制矩形。

Pillow库简介

Pillow是Python图像处理库中比较常用的一个，可以实现图像的读取、显示、变换、滤镜等操作。Pillow已经成为Python 2.x、3.x中最受欢迎的图片处理库之一。

• 安装Pillow：

pip install Pillow

绘制矩形

使用Pillow绘制矩形需要先创建一个ImageDraw对象，然后使用rectangle()函数绘制矩形。

下面是简单的代码示例：

from PIL import Image, ImageDraw

# 创建图像
image = Image.new('RGB', (300, 200), 'white')
# 创建绘制对象
draw = ImageDraw.Draw(image)
# 绘制矩形
draw.rectangle((50, 50, 200, 150), outline='red', width=2)
# 显示图像
image.show()

• Image.new()函数创建一个指定大小和颜色的图像，其中'RGB'表示颜色模式为RGB，(300, 200)表示宽300像素高200像素，'white'表示背景色为白色。
• ImageDraw.Draw()函数创建绘制对象。
• draw.rectangle()函数绘制矩形，其中(50, 50, 200, 150)表示矩形的左上角和右下角坐标，outline='red'表示矩形的颜色为红色，width=2表示矩形的线条宽度为2像素。
• image.show()函数显示图像。

运行代码，可以看到如下的图片：

绘制多个矩形

如果要在一个图像上绘制多个矩形，只需要多次调用draw.rectangle()函数即可。

from PIL import Image, ImageDraw

# 创建图像
image = Image.new('RGB', (300, 200), 'white')
# 创建绘制对象
draw = ImageDraw.Draw(image)
# 绘制多个矩形
draw.rectangle((50, 50, 150, 100), outline='red', width=2)
draw.rectangle((100, 100, 200, 150), outline='green', width=2)
draw.rectangle((150, 50, 250, 150), outline='blue', width=2)
# 显示图像
image.show()

运行代码，可以看到如下的图片：

总结

使用Pillow库在Python中绘制矩形非常方便，只需要创建一个ImageDraw对象，然后使用rectangle()函数绘制矩形即可。如果要绘制多个矩形，只需要多次调用rectangle()函数即可。