如何在PyQt6中实现可多选下拉列表，用于DataFrame多列绘图选择

实现思路 #

Qt6没有原生的多选下拉组件，但可以通过自定义QComboBox来实现：

• 用QStandardItemModel作为下拉框的数据源，给每个Item添加复选框属性
• 重写QComboBox的showPopup()逻辑，确保下拉时所有Item的复选框状态正确
• 实现获取选中列名的方法，方便后续提取DataFrame列进行绘图
• 这种方式既节省界面空间，又比单独罗列复选框性能更优，因为仅在下拉展开时渲染选项

完整代码示例 #

以下是包含DataFrame加载、多选下拉框、绘图功能的完整示例：

import sys
import pandas as pd
import numpy as np
from PyQt6.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout, 
                             QComboBox, QPushButton, QLabel)
from PyQt6.QtGui import QStandardItem, QStandardItemModel
from matplotlib.backends.backend_qtagg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.figure import Figure

class CustomMultiComboBox(QComboBox):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setModel(QStandardItemModel(self))
        self.view().pressed.connect(self.handle_item_pressed)
        self._selected_items = []

    def handle_item_pressed(self, index):
        # 切换选中项的复选框状态
        item = self.model().itemFromIndex(index)
        if item.checkState() == item.CheckState.Checked:
            item.setCheckState(item.CheckState.Unchecked)
            self._selected_items.remove(item.text())
        else:
            item.setCheckState(item.CheckState.Checked)
            self._selected_items.append(item.text())
        # 更新下拉框显示文本
        self.update_display_text()

    def add_items(self, items):
        # 添加选项并设置初始未选中
        self._selected_items = []
        self.model().clear()
        for item_text in items:
            item = QStandardItem(item_text)
            item.setCheckable(True)
            item.setCheckState(item.CheckState.Unchecked)
            self.model().appendRow(item)
        self.update_display_text()

    def update_display_text(self):
        # 根据选中项数量更新显示文本
        if not self._selected_items:
            self.setPlaceholderText("请选择列")
        elif len(self._selected_items) > 3:
            self.setCurrentText(f"已选 {len(self._selected_items)} 列")
        else:
            self.setCurrentText(", ".join(self._selected_items))

    def get_selected_items(self):
        # 返回选中的列名列表
        return self._selected_items

class MainWindow(QMainWindow):
    def __init__(self):
        super().__init__()
        self.setWindowTitle("DataFrame多列绘图工具")
        self.setGeometry(100, 100, 800, 600)

        # 生成示例DataFrame
        self.df = pd.DataFrame({
            "时间": pd.date_range(start="2024-01-01", periods=30),
            "温度": np.random.randn(30).cumsum() + 20,
            "湿度": np.random.randn(30).cumsum() + 60,
            "气压": np.random.randn(30).cumsum() + 1013,
            "风速": np.random.randint(0, 10, size=30),
            "降雨量": np.random.randint(0, 5, size=30)
        })

        # 构建UI
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        layout = QVBoxLayout(central_widget)

        # 多选下拉框
        self.column_combo = CustomMultiComboBox()
        self.column_combo.add_items(self.df.columns[1:])  # 跳过时间列
        layout.addWidget(QLabel("选择要绘制的列："))
        layout.addWidget(self.column_combo)

        # 绘图按钮
        self.plot_btn = QPushButton("绘制图表")
        self.plot_btn.clicked.connect(self.plot_selected_columns)
        layout.addWidget(self.plot_btn)

        # Matplotlib画布
        self.figure = Figure(figsize=(8, 4))
        self.canvas = FigureCanvas(self.figure)
        layout.addWidget(self.canvas)

    def plot_selected_columns(self):
        selected_cols = self.column_combo.get_selected_items()
        if not selected_cols:
            return

        # 清空画布
        self.figure.clear()
        ax = self.figure.add_subplot(111)

        # 绘制选中列
        for col in selected_cols:
            ax.plot(self.df["时间"], self.df[col], label=col)

        ax.set_xlabel("时间")
        ax.set_ylabel("数值")
        ax.legend()
        ax.tick_params(axis='x', rotation=45)
        self.figure.tight_layout()

        # 更新画布
        self.canvas.draw()

if __name__ == "__main__":
    app = QApplication(sys.argv)
    window = MainWindow()
    window.show()
    sys.exit(app.exec())

优化建议 #

• 大数据量适配：如果DataFrame列数极多（比如上百列），建议改用QAbstractListModel实现虚拟模型，仅渲染可见区域的选项，避免一次性加载所有Item导致的卡顿
• 显示优化：当选中列数过多时，下拉框的显示文本可以简化为"已选X列"，避免文本过长溢出
• 快捷操作：可以在下拉菜单顶部添加"全选""反选"选项，提升用户操作效率
• 异步绘图：如果绘图数据量较大，建议将绘图逻辑放到子线程中执行，避免阻塞GUI主线程，导致界面无响应
• 状态记忆：可以保存用户上次的选中状态，下次打开软件时自动恢复

内容的提问来源于stack exchange，提问作者gabriel belmino freitas