【PyQt】使用PyQt5和Matplotlib实现的CSV数据可视化工具

最新推荐文章于 2025-04-27 21:08:09 发布

shanks66

最新推荐文章于 2025-04-27 21:08:09 发布

阅读量682

点赞数 13

分类专栏： PyQt 文章标签：信息可视化 pyqt qt

本文链接：https://blog.csdn.net/weixin_59422604/article/details/145459936

版权

PyQt 专栏收录该内容

43 篇文章

订阅专栏

使用PyQt5和Matplotlib实现的CSV数据可视化工具

界面展示

在这里插入图片描述

代码

import sys
from PyQt5.QtWidgets import (QApplication, QMainWindow, QWidget, QVBoxLayout,
                             QHBoxLayout, QPushButton, QComboBox, QFileDialog,
                             QLabel, QMessageBox)
import pandas as pd
from matplotlib.figure import Figure
from matplotlib.backends.backend_qt5agg import FigureCanvasQTAgg as FigureCanvas
from matplotlib.backends.backend_qt5agg import NavigationToolbar2QT as NavigationToolbar

class CSVVisualizer(QMainWindow):
    def __init__(self):
        super().__init__()
        self.initUI()
        self.df = None  # 用于存储CSV数据

    def initUI(self):
        self.setWindowTitle('CSV数据可视化工具')
        self.setGeometry(100, 100, 1024, 768)

        # 创建中心部件和主布局
        central_widget = QWidget()
        self.setCentralWidget(central_widget)
        main_layout = QVBoxLayout(central_widget)

        # 创建菜单栏
        menubar = self.menuBar()
        file_menu = menubar.addMenu('文件')
        open_action = file_menu.addAction('打开CSV')
        open_action.triggered.connect(self.load_csv)

        # 创建控件容器
        controls = QWidget()
        control_layout = QHBoxLayout(controls)
        
        # 添加控件
        control_layout.addWidget(QLabel("X轴:"))
        self.x_combo = QComboBox()
        control_layout.addWidget(self.x_combo)
        
        control_layout.addWidget(QLabel("Y轴:"))
        self.y_combo = QComboBox()
        control_layout.addWidget(self.y_combo)
        
        self.plot_btn = QPushButton("绘制图表")
        self.plot_btn.clicked.connect(self.plot_chart)
        control_layout.addWidget(self.plot_btn)

        # 添加Matplotlib组件
        self.figure = Figure()
        self.canvas = FigureCanvas(self.figure)
        self.toolbar = NavigationToolbar(self.canvas, self)

        # 将组件添加到主布局
        main_layout.addWidget(controls)
        main_layout.addWidget(self.toolbar)
        main_layout.addWidget(self.canvas)

    def load_csv(self):
        """加载CSV文件"""
        path, _ = QFileDialog.getOpenFileName(
            self, '打开CSV文件', '', 'CSV文件 (*.csv)')
        
        if path:
            try:
                self.df = pd.read_csv(path)
                columns = self.df.columns.tolist()
                
                # 更新下拉菜单
                self.x_combo.clear()
                self.y_combo.clear()
                self.x_combo.addItems(columns)
                self.y_combo.addItems(columns)
                
            except Exception as e:
                QMessageBox.critical(self, '错误', f'文件读取失败: {str(e)}')

    def plot_chart(self):
        """绘制图表"""
        if self.df is None:
            QMessageBox.warning(self, '警告', '请先加载CSV文件')
            return
        
        x_col = self.x_combo.currentText()
        y_col = self.y_combo.currentText()
        
        if not x_col or not y_col:
            QMessageBox.warning(self, '警告', '请选择X/Y轴列')
            return
        
        try:
            # 清除旧图表
            self.figure.clear()
            ax = self.figure.add_subplot(111)
            
            # 绘制散点图
            ax.scatter(self.df[x_col], self.df[y_col], alpha=0.6)
            ax.set_xlabel(x_col)
            ax.set_ylabel(y_col)
            ax.set_title(f'{x_col} vs {y_col}')
            
            # 自动调整布局并刷新
            self.figure.tight_layout()
            self.canvas.draw()
            
        except Exception as e:
            QMessageBox.critical(self, '错误', f'绘图失败: {str(e)}')

if __name__ == '__main__':
    app = QApplication(sys.argv)
    window = CSVVisualizer()
    window.show()
    sys.exit(app.exec_())