PyQt实现按钮触发的带参数多矩形绘制功能

一、基于QPainter的实现方案 #

Qt中自定义绘图逻辑必须放在paintEvent函数中执行，这是Qt绘图机制的核心要求。我们可以通过自定义Widget承载绘图区域，配合按钮触发数据更新与重绘，满足你的两个需求：

1. 点击按钮绘制单个矩形 #

• 核心逻辑：自定义Widget维护单个矩形的坐标数据，按钮点击时设置该坐标，调用update()触发重绘，paintEvent中用QPainter绘制矩形。
• 代码示例（PyQt5）：

from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QVBoxLayout
from PyQt5.QtGui import QPainter, QPen
from PyQt5.QtCore import Qt

class DrawWidget(QWidget):
    def __init__(self):
        super().__init__()
        self.single_rect = None  # 存储单个矩形的(x1,y1,x2,y2)

    def paintEvent(self, event):
        super().paintEvent(event)
        painter = QPainter(self)
        painter.setPen(QPen(Qt.red, 2))  # 设置画笔样式
        if self.single_rect:
            x1, y1, x2, y2 = self.single_rect
            # 转换为Qt drawRect所需的左上角坐标+宽高格式
            painter.drawRect(min(x1,x2), min(y1,y2), abs(x2-x1), abs(y2-y1))

class MainWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.init_ui()

    def init_ui(self):
        layout = QVBoxLayout()
        self.draw_widget = DrawWidget()
        self.draw_widget.setFixedSize(400, 300)  # 固定绘图区域大小
        btn = QPushButton("绘制单个矩形")
        btn.clicked.connect(self.draw_single_rect)
        layout.addWidget(self.draw_widget)
        layout.addWidget(btn)
        self.setLayout(layout)

    def draw_single_rect(self):
        # 这里可以替换为任意你需要的坐标（比如用户输入、随机生成）
        self.draw_widget.single_rect = (50, 50, 200, 150)
        self.draw_widget.update()  # 触发重绘

if __name__ == "__main__":
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()

2. 传递矩形列表绘制多矩形 #

• 核心逻辑：自定义Widget维护一个矩形列表，通过接口传入包含(x1,y1,x2,y2)的元组列表，调用update()触发重绘，paintEvent中遍历列表逐个绘制。
• 代码示例（基于上述DrawWidget修改）：

class DrawWidget(QWidget):
    def __init__(self):
        super().__init__()
        self.single_rect = None
        self.rect_list = []  # 存储多矩形的列表

    def set_rect_list(self, rects):
        self.rect_list = rects
        self.update()

    def paintEvent(self, event):
        super().paintEvent(event)
        painter = QPainter(self)
        painter.setPen(QPen(Qt.red, 2))
        # 绘制单个矩形
        if self.single_rect:
            x1, y1, x2, y2 = self.single_rect
            painter.drawRect(min(x1,x2), min(y1,y2), abs(x2-x1), abs(y2-y1))
        # 批量绘制多矩形
        for rect in self.rect_list:
            x1, y1, x2, y2 = rect
            painter.drawRect(min(x1,x2), min(y1,y2), abs(x2-x1), abs(y2-y1))

在MainWindow中添加多矩形触发逻辑：

def init_ui(self):
    # ... 原有代码 ...
    multi_btn = QPushButton("绘制多矩形")
    multi_btn.clicked.connect(self.draw_multi_rects)
    layout.addWidget(multi_btn)

def draw_multi_rects(self):
    # 示例矩形列表，可替换为外部传入的数据
    rects = [(30, 30, 100, 80), (150, 100, 250, 200), (280, 50, 350, 150)]
    self.draw_widget.set_rect_list(rects)

如果使用C++ Qt，逻辑完全一致：自定义QWidget子类，重写paintEvent，维护矩形数据成员，按钮点击信号触发数据更新并调用update()即可。

二、替代QPainter的方案：QGraphicsView/QGraphicsScene框架 #

如果你的项目涉及大量图形元素的管理、交互（比如移动、删除矩形），QGraphicsView框架会比直接用QPainter更便捷，它内置了图形项的管理机制，无需手动处理重绘逻辑：

• 核心逻辑：创建QGraphicsScene作为画布，QGraphicsView作为视图显示场景；按钮点击时，直接向场景中添加QGraphicsRectItem（单个或批量）即可。
• 代码示例（PyQt5）：

from PyQt5.QtWidgets import QApplication, QWidget, QPushButton, QVBoxLayout, QGraphicsView, QGraphicsScene, QGraphicsRectItem
from PyQt5.QtGui import QPen
from PyQt5.QtCore import Qt

class MainWindow(QWidget):
    def __init__(self):
        super().__init__()
        self.init_ui()

    def init_ui(self):
        layout = QVBoxLayout()
        # 创建场景和视图
        self.scene = QGraphicsScene()
        self.view = QGraphicsView(self.scene)
        self.view.setFixedSize(400, 300)
        # 单个矩形按钮
        single_btn = QPushButton("绘制单个矩形")
        single_btn.clicked.connect(self.add_single_rect)
        # 多矩形按钮
        multi_btn = QPushButton("绘制多矩形")
        multi_btn.clicked.connect(self.add_multi_rects)
        layout.addWidget(self.view)
        layout.addWidget(single_btn)
        layout.addWidget(multi_btn)
        self.setLayout(layout)

    def add_single_rect(self):
        # 创建矩形项，参数为x,y,width,height
        rect_item = QGraphicsRectItem(50,50,150,100)
        rect_item.setPen(QPen(Qt.red,2))
        self.scene.addItem(rect_item)

    def add_multi_rects(self):
        rects = [(30,30,70,50), (150,100,100,100), (280,50,70,100)]
        for x1,y1,x2,y2 in rects:
            rect_item = QGraphicsRectItem(min(x1,x2), min(y1,y2), abs(x2-x1), abs(y2-y1))
            rect_item.setPen(QPen(Qt.red,2))
            self.scene.addItem(rect_item)

if __name__ == "__main__":
    app = QApplication([])
    window = MainWindow()
    window.show()
    app.exec_()

这个方案的优势在于：图形项的添加、删除、移动操作更简单，支持碰撞检测、分层显示等高级特性，适合复杂的图形交互场景。

内容的提问来源于stack exchange，提问作者brownfox2k6