web-dev-qa-db-ja.com

PyQt-QTableViewで表示するためのQAbstractTableModelを実装する

pandasデータフレームをPyQtテーブルに表示します。これである程度の進歩はありましたが、テーブルモデルクラスを正しく派生させることができませんでした。これに関するヘルプはとても有難い。

**完全なコード例に注意してください ここ **

有効なQtCore.QAbstractTableModel派生クラスを生成するのに苦労しています。 QItemDelegatesに関する前の質問に続いて、Pandas DataFrameから実際のデータを挿入するためにテーブルモデルを生成しようとしています。実際のサンプルコード here がありますが、ウィジェットクラスのTableModel2でTableModelを置き換えました（ln 152）テーブルを表示できません。

class TableModel2(QtCore.QAbstractTableModel): 
    def __init__(self, parent=None, *args): 
        super(TableModel2, self).__init__()
        #QtCore.QAbstractTableModel.__init__(self, parent, *args)
        self.datatable = None
        self.headerdata = None
        self.dataFrame = None
        self.model = QtGui.QStandardItemModel(self)

    def update(self, dataIn):
        print 'Updating Model'
        self.datatable = dataIn
        print 'Datatable : {0}'.format(self.datatable)
        headers = dataIn.columns.values
        header_items = [
                    str(field)
                    for field in headers
        ]
        self.headerdata = header_items
        print 'Headers'
        print self.headerdata

        for i in range(len(dataIn.index.values)):
            for j in range(len(dataIn.columns.values)):
                #self.datatable.setItem(i,j,QtGui.QTableWidgetItem(str(df.iget_value(i, j))))
                self.model.setItem(i,j,QtGui.QStandardItem(str(dataIn.iget_value(i, j))))

    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 not index.isValid(): 
            return QtCore.QVariant()
        Elif role != QtCore.Qt.DisplayRole: 
            return QtCore.QVariant() 
        #return QtCore.QVariant(self.model.data(index)) 
            return QtCore.QVariant(self.model.data(index)) 

    def headerData(self, col, orientation, role):
        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
            return QtCore.QVariant()
        return QtCore.QVariant(self.headerdata[col])

    def setData(self, index, value, role=QtCore.Qt.DisplayRole):
        print "setData", index.row(), index.column(), value

    def flags(self, index):
        if (index.column() == 0):
            return QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsEnabled
        else:
            return QtCore.Qt.ItemIsEnabled

次のように、モデルを作成してビューに追加しようとしています。

class Widget(QtGui.QWidget):
    """
    A simple test widget to contain and own the model and table.
    """
    def __init__(self, parent=None):
        QtGui.QWidget.__init__(self, parent)

        l=QtGui.QVBoxLayout(self)
        cdf = self.get_data_frame()
        self._tm=TableModel(self)
        self._tm.update(cdf)
        self._tv=TableView(self)
        self._tv.setModel(self._tm)
        for row in range(0, self._tm.rowCount()):
            self._tv.openPersistentEditor(self._tm.index(row, 0))
        l.addWidget(self._tv)

    def get_data_frame(self):
        df = pd.DataFrame({'Name':['a','b','c','d'], 
        'First':[2.3,5.4,3.1,7.7], 'Last':[23.4,11.2,65.3,88.8], 'Class':[1,1,2,1], 'Valid':[True, True, True, False]})
        return df

ご清聴ありがとうございました！

注：Edit 2 QStandardItemModelをTableModel2に組み込みました。 @mataのコメントの後のdataFrameToQtTable関数も削除しました。これは少し近づいていますが、まだ機能していません。

pythonqtqt4pyqt
16
drexiya

OKこれは、上記の提案と、SummerfieldによるRapid GUIブックからの助けを借りて、これを理解しました。 QAbstractTableModelに存在する基本モデルはありません。オーバーライドする必要がある関数は3つだけです。データは、データ呼び出しで返される限り、ユーザー定義の形式で保存できます。

非常に簡単な実装は次のとおりです。

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

    def update(self, dataIn):
        print 'Updating Model'
        self.datatable = dataIn
        print 'Datatable : {0}'.format(self.datatable)

    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()

    def flags(self, index):
        return QtCore.Qt.ItemIsEnabled

これにより、Qtビューで互換性のあるデータフレームを表示できます。

要点を更新しました ここ

これにより、これも行う必要がある場合は、すぐに実行できます。

15
drexiya

これはおそらくあなたの問題です：

def rowCount(self, parent=QtCore.QModelIndex()):
    if type(self.datatable) == pd.DataFrame:
    ...


def columnCount(self, parent=QtCore.QModelIndex()):
    if (self.datatable) == pd.DataFrame:
    ...

datatableをQTableWidgetのdataFrameToQtTableに設定したので、pd.DataFrame、メソッドは常に0を返します。

型チェックがなければ、すぐに問題が発生します。型が一致しないすべてのケースを黙って無視してもよろしいですか（予期しているのと同じインターフェイスに従っていない場合は、エラーを発生させてください）？タイプチェックは ほとんどの場合不要 です。

2
mata

Pandas 0.13は実験的な機能として提供します：

Qtpandas DataFrameModelおよびDataFrameWidgetのPySideサポート

https://github.com/pydata/pandas/blob/master/doc/source/faq.rst を参照してください

この機能を使用して追加できます

from pandas.sandbox.qtpandas import DataFrameModel, DataFrameWidget
1
working4coins