PyQt显示tableframe垂直标题的数据帧

时间:2018-01-24 09:51:10

标签: python python-3.x pandas pyqt pyqt5

我想知道如何在tableview中显示垂直标题。垂直标题应显示索引。

我创建了一个股票价格数据框。日期是数据帧的索引(即2017-12-06),收盘价是数据帧的值(即50 $)。

我写了一个使用QAbstractTabelModel的模型类。 我的headerData方法如下:

def headerData(self, rowcol, orientation, role):
    if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
        return self._data.columns[rowcol]
    if orientation == QtCore.Qt.Vertical and role == QtCore.Qt.DisplayRole:
        return self._data.index[rowcol]
    return None

但显示的tableview中没有索引。

显示的TableView的屏幕截图

整个源代码如下:

from PyQt5 import QtCore

class PandasModel(QtCore.QAbstractTableModel):

    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, rowcol, orientation, role):
        if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
            return self._data.columns[rowcol]
        if orientation == QtCore.Qt.Vertical and role == QtCore.Qt.DisplayRole:
            return self._data.index[rowcol]
        return None

我有来自stackoverflow reference

的代码

编辑:我的数据框源代码附件

dates = pd.date_range(start_date, end_date)
df = pd.DataFrame(index=dates)
df_temp = pd.read_csv(directory, index_col='Date', parse_dates=True, usecols=['Date', 'Adj Close'],
                      na_values=['nan'])
df = df.join(df_temp)
df = df.dropna(axis=0, how='any')
return df

编辑2:

csv文件的屏幕截图

1 个答案:

答案 0 :(得分:0)

问题是由于索引返回PyQt无法识别的数据类型Timestamp引起的,解决方法是将其转换为字符串,为此我们有两个可能的选项:使用str()或方法strftime()表示格式:

def headerData(self, rowcol, orientation, role=QtCore.Qt.DisplayRole):
    if orientation == QtCore.Qt.Horizontal and role == QtCore.Qt.DisplayRole:
        return self._data.columns[rowcol]
    if orientation == QtCore.Qt.Vertical and role == QtCore.Qt.DisplayRole:
        return self._data.index[rowcol].strftime('%Y-%m-%d') # or str(self._data.index[rowcol])