Question

我试图实现一个Model，以便在通过QT Designer设计的tableView对象中显示来自pandas数据帧的数据。我在Stack Overflow中看到了一些示例，展示了如何为此目的编写一个简单的模型：

import sip
sip.setapi('QString', 1)
sip.setapi('QVariant', 1)

import pandas as pd
from PyQt4 import QtCore, QtGui


class TableModel(QtCore.QAbstractTableModel): 
    def __init__(self, parent=None, *args): 
        super(TableModel, self).__init__()
        self.datatable = None     

    def update(self, dataIn):
        self.datatable = dataIn            

    def rowCount(self, parent=QtCore.QModelIndex()):
        return len(self.datatable.index) 

    def columnCount(self, parent=QtCore.QModelIndex()):
        return len(self.datatable.columns.values) 

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role == QtCore.Qt.DisplayRole:            
            i = index.row()
            j = index.column()
            return '{0}'.format(self.datatable.iget_value(i, j))
        else:
            return QtCore.QVariant()

但是，我无法成功编写TableModel类中的任何函数，它允许我编辑水平和垂直标题，以便显示数据框索引（在行上）和dataframe列标签（在表的列上）。任何人都可以帮助我吗？谢谢

Answer 1

我实际上发现这个小部件产生了我希望的结果：

def headerData(self,section,orientation,role=Qt.DisplayRole):
    if role != Qt.DisplayRole:
        return QVariant()          
    if orientation == Qt.Horizontal:
        return self.df.columns.tolist()[section]
    elif orientation == Qt.Vertical:
        return str(self.df.index.tolist()[section])

任何其他更有效的想法？

Pandas DataFrame的自定义tableView模型

1 个答案: