Я пытался выяснить, как отобразить подмножество таблицы, используя виджет QTableView.У меня странная ошибка, и я не уверен, что это ошибка или проблема с моим кодом.Я могу загрузить в набор данных просто отлично.Когда программа запускается и нажимается кнопка загрузки данных, QTableView отображает правильное количество строк и имен столбцов, но большинство, если не все, данных отсутствуют.Однако данные появятся, если вы попытаетесь отсортировать таблицу, щелкнув любое из имен столбцов.Некоторые сообщения об ошибках при первой загрузке в набор данных говорят: «# у нас по определению только 0-я ось» или «# мы получили скаляр?».
Я подумал, что это может быть проблема сCSV-файл, который я использовал, но я смог легко воспроизвести его с помощью рандомизированной таблицы чисел.Комментирование функции сортировки по-прежнему не приводит к отображению данных, но теперь сортировка не может использоваться для отображения данных.Код, который я использовал в моем примере, является измененным ответом здесь .
Импорт необходимых пакетов и функций.
from PyQt5 import QtCore, QtGui, QtWidgets
import pandas as pd
import numpy as np
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.layoutAboutToBeChanged.emit()
self._df.sort_values(colname, ascending= order == QtCore.Qt.AscendingOrder, inplace=True)
self._df.reset_index(inplace=True, drop=True)
self.layoutChanged.emit()
Приложение, которое отображает таблицу и обрабатываетsubsetting.
class Widget(QtWidgets.QWidget):
def __init__(self, parent=None):
QtWidgets.QWidget.__init__(self, parent=None)
vLayout = QtWidgets.QVBoxLayout(self)
hLayout = QtWidgets.QHBoxLayout()
self.loadBtn = QtWidgets.QPushButton("Load Table", self)
hLayout.addWidget(self.loadBtn)
vLayout.addLayout(hLayout)
self.pandasTv = QtWidgets.QTableView(self)
vLayout.addWidget(self.pandasTv)
self.loadBtn.clicked.connect(self.loadTable)
#self.pandasTv.setSortingEnabled(True)
def loadTable(self):
dfNum = pd.DataFrame(np.random.randint(0,100,size=(20, 4)), columns=list('ABCD'))
dfNumSub = dfNum.loc[dfNum['A'].astype(str).str.contains('1')]
model = PandasModel(dfNumSub)
self.pandasTv.setModel(model)
if __name__ == "__main__":
import sys
app = QtWidgets.QApplication(sys.argv)
w = Widget()
w.show()
sys.exit(app.exec_())
Я хочу, чтобы все найденные данные отображались без сортировки столбцов.
EDIT: эта ошибка возникает только при поднаборе кадра данных.Если вместо этого изменить код для отображения dfNum, все данные будут отображаться очень хорошо.