数据可视化 第4章



  • 第4章 添加表格QTableView

    1.添加table_model.py,里面子类化QAbstractTableModel,实现自定义table model

    from PySide2.QtCore import Qt, QAbstractTableModel, QModelIndex
    from PySide2.QtGui import QColor
    
    #实现CustromTableModel类
    class CustomTableModel(QAbstractTableModel):
        def __init__(self, data=None):
            QAbstractTableModel.__init__(self)
            self.load_data(data)
    
        def load_data(self, data):
            self.input_datas = data[0].values
            self.input_magnitudes = data[1].values
    
            self.column_count = 2
            self.row_count = len(self.input_magnitudes)
    
        def rowCount(self, parent=QModelIndex()):
            return self.row_count
    
        def columnCount(self, parent=QModelIndex()):
                return self.column_count
    
        def headerData(self, section, orientation, role):
            if role != Qt.DisplayRole:
                return None
            if orientation == Qt.Horizontal:
                return ("Date", "Magnitude")[section]
            else:
                return "{}".format(section)
    
        def data(self, index, role=Qt.DisplayRole):
            column = index.column()
            row = index.row()
    
            if role == Qt.DisplayRole:
                if column == 0:
                    raw_data = self.input_datas[row]
                    data = "{}".format(raw_data.toPython())
                    return data[:-3]
                elif column == 1:
                    return "{:.2f}".format(self.input_magnitudes[row])
            elif role == Qt.BackgroundRole:
               return QColor(Qt.white)
            elif role == Qt.TextAlignmentRole:
               return Qt.AlignRight
    
            return None
    
    

    2.添加main_widget.py,使用QTableView+TableModel显示表格,设置widget布局,自适应窗口

    from PySide2.QtWidgets import (QHBoxLayout, QHeaderView, QSizePolicy, QTableView, QWidget)
    from table_model import *
    
    #含TableView的窗口类
    class Widget(QWidget):
        def __init__(self, data):
            QWidget.__init__(self)
    
            #获取model
            self.model = CustomTableModel(data)
    
            #创建QTableView
            self.table_view = QTableView()
            self.table_view.setModel(self.model)
    
            #QTableView 表头自适应以及拉伸
            self.horizontal_header = self.table_view.horizontalHeader()
            self.vertical_header = self.table_view.verticalHeader()
            self.horizontal_header.setSectionResizeMode(QHeaderView.ResizeToContents)
            self.vertical_header.setSectionResizeMode(QHeaderView.ResizeToContents)
            self.horizontal_header.setStretchLastSection(True)
    
            #布局
            self.main_layout = QHBoxLayout()
            size = QSizePolicy(QSizePolicy.Preferred, QSizePolicy.Preferred)
    
            #应用size策略
            size.setHorizontalStretch(1)
            self.table_view.setSizePolicy(size)
            self.main_layout.addWidget(self.table_view)
    
            #应用布局
            self.setLayout(self.main_layout)
    
    

    3.添加main_window.py,在主窗口里使用定制好的widget

    from PySide2.QtCore import Slot, qApp
    from PySide2.QtGui import QKeySequence
    from PySide2.QtWidgets import QMainWindow, QAction, QApplication
    
    #主窗口
    class MainWindow(QMainWindow):
        def __init__(self, widget):
            QMainWindow.__init__(self)
            self.setWindowTitle("地震信息")
            self.setCentralWidget(widget)
    
            # Menu
            self.menu = self.menuBar()
            self.file_menu = self.menu.addMenu("文件")
    
            # Exit QAction
            exit_action = QAction("退出", self)
            exit_action.setShortcut(QKeySequence.Quit)
            exit_action.triggered.connect(self.close)
    
            self.file_menu.addAction(exit_action)
    
            # Status Bar
            self.status = self.statusBar()
            self.status.showMessage("数据加载和绘制")
    
            # Window dimensions
            geometry = qApp.desktop().availableGeometry(self)
            self.setFixedSize(geometry.width() * 0.6, geometry.height() * 0.4)
    
    

    4.添加chapter4.py,这是我们要运行的py文件,读取命令行参数,解析csv文件,将data数据传递给窗口显示

    import sys, argparse
    import pandas as pd
    from PySide2.QtCore import QDateTime, QTimeZone
    
    from main_widget import Widget
    from main_window import MainWindow
    
    from PySide2.QtWidgets import QApplication
    
    def transform_date(utc, timezone=None):
        utc_fmt = "yyyy-MM-ddTHH:mm:ss.zzzZ"
        new_date = QDateTime().fromString(utc, utc_fmt)
    #    print (new_date.toString(utc_fmt))
        if timezone:
            new_date.setTimeZone(timezone)
    
        return new_date
    
    def read_data(fname):
        df = pd.read_csv(fname)
    
        #移除错误的mag
        df = df.drop(df[df.mag < 0].index)
        magnitudes = df["mag"]
    
        #本地时区
        timezone = QTimeZone(b"Asia/Beijing")
    
        #将时间戳转换为本地时区
        times = df["time"].apply(lambda x: transform_date(x, timezone))
    
        return times, magnitudes
    
    
    
    if __name__ == "__main__":
        options = argparse.ArgumentParser()
        options.description = "窗口显示csv信息"
        options.add_argument("-f", "--file", type=str, required=True, help="CSV文件路径")
        args = options.parse_args()
        data = read_data(args.file)
    
    #    print (data)
        #qt 应用
        app = QApplication(sys.argv)
    
        widget = Widget(data)
        window = MainWindow(widget)
        window.show()
    
        sys.exit(app.exec_())
    
    

    5.运行效果
    4.png



  • QTableView还是比较成熟的,因为很简单。但是如果是树状图,那么只能用QAbstractItemModel的子类了。因为有父子层级关系,所以要自己写。


Log in to reply
 

最近的回复

  • G

    北京名律免费法律咨询,解决您的法律困扰,如果您眼下没有遇到法律上的问题,也可以留一位大律师的联系方式以备不时之需!ccfd91efc3a6b2ed0e79054d2248eed.jpg bf6015b383483f1fe83cdcfd130fc6b.jpg 2f347879129add1096bf3424edbe517.jpg

    read more
  • G

    北京名律免费法律咨询,解决您的法律困扰,如果您眼下没有遇到法律上的问题,也可以留一位大律师的联系方式以备不时之需!微信图片_20200210141336.png 微信图片_20200210141333.png 微信图片_20200210141329.png

    read more
  • G

    北京名律免费法律咨询,解决您的法律困扰,如果您眼下没有遇到法律上的问题,也可以留一位大律师的联系方式以备不时之需!

    read more

关注我们

微博
QQ群