在我的 gui 文件中,我创建了一个 QTableView,如下所示(这是由 Qt Designer 自动生成的部分):
self.pnl_results = QtGui.QTableView(self.tab_3)
font = QtGui.QFont()
font.setPointSize(7)
self.pnl_results.setFont(font)
self.pnl_results.setFrameShape(QtGui.QFrame.StyledPanel)
self.pnl_results.setFrameShadow(QtGui.QFrame.Sunken)
self.pnl_results.setEditTriggers(QtGui.QAbstractItemView.AllEditTriggers)
self.pnl_results.setShowGrid(True)
self.pnl_results.setSortingEnabled(True)
self.pnl_results.setCornerButtonEnabled(True)
self.pnl_results.setObjectName("pnl_results")
然后我定义了一个模型,使我能够将 pandas 数据框链接到 QTableView:
class PandasModel(QtCore.QAbstractTableModel):
"""
Class to populate a table view with a pandas dataframe
"""
def __init__(self, data, parent=None):
QtCore.QAbstractTableModel.__init__(self, parent)
self._data = data
def rowCount(self, parent=None):
return len(self._data.values)
def columnCount(self, parent=None):
return self._data.columns.size
def data(self, index, role=QtCore.Qt.DisplayRole):
if index.isValid():
if role == QtCore.Qt.DisplayRole:
return str(self._data.values[index.row()][index.column()])
return None
def headerData(self, col, orientation, role):
if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
return self._data.columns[col]
return None
def setData(self, index, value, role):
if not index.isValid():
return False
if role != QtCore.Qt.EditRole:
return False
row = index.row()
if row < 0 or row >= len(self._data.values):
return False
column = index.column()
if column < 0 or column >= self._data.columns.size:
return False
self._data.values[row][column] = value
self.dataChanged.emit(index, index)
return True
def flags(self, index):
flags = super(self.__class__,self).flags(index)
flags |= QtCore.Qt.ItemIsEditable
flags |= QtCore.Qt.ItemIsSelectable
flags |= QtCore.Qt.ItemIsEnabled
flags |= QtCore.Qt.ItemIsDragEnabled
flags |= QtCore.Qt.ItemIsDropEnabled
return flags
最后将我的 pandas 数据框 (df) 添加到模型中:
model = PandasModel(df)
self.ui.pnl_results.setModel(model)
这会在 QTableView 中正确显示我的 pandas 数据框。但是,出于某种原因,当我编辑字段时,返回到它们的原始值(而且一旦我编辑字段,它就开始为空)。如何使其可编辑,然后将结果写回 pandas 数据框?
最佳答案
您的模型缺少 setData方法。 QtCore.QAbstractTableModel
的默认实现不执行任何操作并返回 False
。您需要在您的模型中实现此方法以使其项目可编辑。如果 df
是存储数据的实际容器,您只需在 setData
方法中更改存储在容器中的项目的值。它可能看起来像这样:
def setData(self, index, value, role):
if not index.isValid():
return False
if role != QtCore.Qt.EditRole:
return False
row = index.row()
if row < 0 or row >= len(self._data.values):
return False
column = index.column()
if column < 0 or column >= self._data.columns.size:
return False
self._data.values[row][column] = value
self.dataChanged.emit(index, index)
return True
您还需要实现 flags方法返回包含 QtCore.Qt.ItemIsEditable
的值。
关于python - 当模型是 pandas dataframe 时使 QTableView 可编辑,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41192293/