在列设置为索引的PyQt5中显示TableView中的pandas DataFrame

from PyQt5.QtCore import Qt, QAbstractTableModel, QVariant
from PyQt5.QtGui import QColor

class DataFrameTableModel(QAbstractTableModel):
    def __init__(self, data):
        super().__init__()
        self._data = data

    def rowCount(self, parent=None):
        return len(self._data)

    def columnCount(self, parent=None):
        return len(self._data.columns)

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

        row = index.row()
        col = index.column()

        if role == Qt.DisplayRole:
            return str(self._data.iloc[row, col])
        elif role == Qt.BackgroundRole:
            # 可选：设置单元格的背景颜色
            return QColor(Qt.lightGray)

        return QVariant()

    def headerData(self, section, orientation, role=Qt.DisplayRole):
        if role == Qt.DisplayRole:
            if orientation == Qt.Horizontal:
                return str(self._data.columns[section])
            elif orientation == Qt.Vertical:
                return str(self._data.index[section])

        return QVariant()

import sys
from PyQt5.QtWidgets import QApplication, QTableView
import pandas as pd

if __name__ == '__main__':
    app = QApplication(sys.argv)

    # 创建一个示例DataFrame
    data = {'Name': ['Alice', 'Bob', 'Charlie'],
            'Age': [25, 30, 35],
            'City': ['New York', 'Paris', 'London']}
    df = pd.DataFrame(data)

    # 创建TableModel并设置为TableView的模型
    model = DataFrameTableModel(df)
    table_view = QTableView()
    table_view.setModel(model)

    table_view.show()

    sys.exit(app.exec_())