首页 > 解决方案 > 在 Python 中使用 QtGui.QPainter() 更改点对象的颜色

问题描述

我正在使用 qtpy 的几个模块在 Python 中开发地图可视化程序。有一个主窗口界面,它在屏幕上显示包含多个地理定位点的背景地图。每个点的位置由外部 .csv 文件确定,该文件包含有关纬度、经度和其他文本属性的信息。每次实例化地图窗口时,程序都会读入该文件。打开地图窗口时,每个点的颜色默认为红色,但我希望每个点根据其存储在 .csv 文件中的元数据更改为不同的颜色。例如,文件中有一个名为“color”的标题,每个点都有编码的文本字符串“red”、“green”或“blue”。这是到目前为止我一直在处理的代码部分......

# Initialize all points to default color.
color = QtCore.Qt.red

for i, p in zip(range(len(self.points)), self.points):
    if lb_lat <= stn_lat and stn_lat <= ub_lat and window_rect.contains(*self.transform.map(stn_x, stn_y)):
        if p['color'] == 'green':
            color = QtCore.Qt.green
        elif p['color'] == 'blue':
            color = QtCore.Qt.blue
        elif p['color'] == 'red':
            color = QtCore.Qt.red
        else:
            color = QtCore.Qt.white
        qp.setPen(QtGui.QPen(color, self.scale))
        qp.setBrush(QtGui.QBrush(color))
        qp.drawEllipse(QtCore.QPointF(stn_x, stn_y), size, size)

点列表存储在变量 self.points 中,我正在尝试遍历此列表并使用 QtGui.QPen 和 QBrush 将正确的颜色应用于每个点。发生的情况是,如果点 1 的 .csv 文件中的颜色属性具有文本字符串“green”,那么整个点数组将变为绿色,而不仅仅是那个点。查看 if...else 语句之后的代码,我无法找到一种方法来“索引”setPen 和 setBrush 命令以解决问题。着色方法作为一个不可分割的单元作用于整个点阵列,而不是按预期单独处理每个点。有人可能知道使用 Qt 框架的方法吗?请让我知道提供额外的代码是否有助于澄清问题或提供更好的上下文,因为我很乐意这样做。

标签: pythonpyqtpyqt5

解决方案


我能够通过删除我在 self.points 中迭代项目的循环结构来解决我遇到的问题。我已经有一个更高级别的“for”循环,这导致每次将点绘制到屏幕上时都会引用不正确的数组索引。现在每个点都更改为适当的颜色。


推荐阅读