使用熊猫将行插入数据后,QTableView意外左移

时间:2018-07-12 00:41:04

标签: python pandas pyqt qtableview

----问题的样子----

MDN equality comparisons and sameness

点击 记录按钮 后,它变成了

enter image description here

如您所见,表格刚刚向左移动,隐藏了垂直标题。

----相关代码----

我的tableivew称为 tableViewTransaction ,它使用模型链接到熊猫的 dataframe 格式的数据 。这是我在 QMainWindow __init__函数内部将它们连接在一起的方式。

    self.data = pd.read_csv('transactions.txt', sep='\t', header=None)
    self.data.columns = ["Name", "Price", "Action"]
    self.model = PandasModel(self.data)
    self.tableViewTransaction.setModel(self.model)

这是我的TableView模型

class PandasModel(QtCore.QAbstractTableModel): 
    def __init__(self, df = pd.DataFrame(), parent=None): 
        QtCore.QAbstractTableModel.__init__(self, parent=parent)
        self._df = df

    def headerData(self, section, orientation, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return QtCore.QVariant()

        if orientation == QtCore.Qt.Horizontal:
            try:
                return self._df.columns.tolist()[section]
            except (IndexError, ):
                return QtCore.QVariant()
        elif orientation == QtCore.Qt.Vertical:
            try:
                # return self.df.index.tolist()
                return self._df.index.tolist()[section]
            except (IndexError, ):
                return QtCore.QVariant()

    def data(self, index, role=QtCore.Qt.DisplayRole):
        if role != QtCore.Qt.DisplayRole:
            return QtCore.QVariant()

        if not index.isValid():
            return QtCore.QVariant()

        return QtCore.QVariant(str(self._df.ix[index.row(), index.column()]))

    def setData(self, index, value, role):
        row = self._df.index[index.row()]
        col = self._df.columns[index.column()]
        if hasattr(value, 'toPyObject'):
            # PyQt4 gets a QVariant
            value = value.toPyObject()
        else:
            # PySide gets an unicode
            dtype = self._df[col].dtype
            if dtype != object:
                value = None if value == '' else dtype.type(value)
        self._df.set_value(row, col, value)
        return True

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

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

    def sort(self, column, order):
        colname = self._df.columns.tolist()[column]
        self.layoutChanged.emit()
        self._df.sort_values(colname, ascending= order == QtCore.Qt.AscendingOrder, inplace=True)
        self._df.reset_index(inplace=True, drop=True)
        self.layoutChanged.emit()

按下 录制按钮 时,将调用以下功能:

    def recordTransaction(self):
        name = self.comboBoxStock.currentText()
        price = self.lineEditMoney.text()
        action = self.comboBoxAction.currentText()

        self.data.loc[len(self.data.index)] = [name, price, action]
        self.tableViewTransaction.model().layoutChanged.emit()

我知道上面的示例中的名称,价格和操作存储的是正确的信息,即上面的示例中的“ stock1”,“ 2”,“ Buy”。

----完整代码----

enter image description here

使用方法: 首先将其解压缩,然后使用python3运行StockSim.py。 在Mac上,使用终端转到 Stock Sim 目录后,只需运行python3 StockSim.py。确保首先安装了python3和pyqt5。

----------------------

我想要的是TableView不会向左移动,非常感谢您的帮助!

0 个答案:

没有答案