PyQt - 在TableView中更改行和单元格的颜色

Question

我有一个带有三列的QTableView，第二列关于数字，只有三种类型：1、-1和0。

我想为这三种“类型”的数字（1、-1、0）设置不同的颜色，使用不同的颜色来渲染它们所在的行。我该如何做？

 self.tableView = QTableView(self.tabSentimento)
 self.tableView.setGeometry(QRect(550,10,510,700))
 self.tableView.setObjectName(_fromUtf8("TabelaSentimento"))
 self.tableView.setModel(self.model)
 self.tableView.horizontalHeader().setStretchLastSection(True)

obs: 我使用了 horizontalheader().setStrechLastSection(True)，因为我通过一个按钮打开一个已经存在的csv文件，并将其加载到我的表格视图中。

- Marco

Answer 1

您需要在模型中定义颜色，而不是在视图中定义：

您需要在模型中定义颜色，而不是在视图中定义：

def data(self, index, role):
    ...
    if role == Qt.BackgroundRole:
        return QBrush(Qt.yellow)

编辑： 这里有一个完整的例子，除了颜色部分完全从http://www.saltycrane.com/blog/2007/06/pyqt-42-qabstracttablemodelqtableview/抄袭而来。

from PyQt4.QtCore import *
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import sys

my_array = [['00','01','02'],
            ['10','11','12'],
            ['20','21','22']]

def main():
    app = QApplication(sys.argv)
    w = MyWindow()
    w.show()
    sys.exit(app.exec_())

class MyWindow(QTableView):
    def __init__(self, *args):
        QTableView.__init__(self, *args)

        tablemodel = MyTableModel(my_array, self)
        self.setModel(tablemodel)

class MyTableModel(QAbstractTableModel):
    def __init__(self, datain, parent=None, *args):
        QAbstractTableModel.__init__(self, parent, *args)
        self.arraydata = datain

    def rowCount(self, parent):
        return len(self.arraydata)

    def columnCount(self, parent):
        return len(self.arraydata[0])

    def data(self, index, role):
        if not index.isValid():
            return QVariant()
        # vvvv this is the magic part
        elif role == Qt.BackgroundRole:
            if index.row() % 2 == 0:
                return QBrush(Qt.yellow)
            else:
                return QBrush(Qt.red)
        # ^^^^ this is the magic part
        elif role != Qt.DisplayRole:
            return QVariant()
        return QVariant(self.arraydata[index.row()][index.column()])

if __name__ == "__main__":
    main()

- rainer

---

首先，谢谢你。但是，你能给我一个例子吗？比如创建一个带有两行不同颜色的表格视图？我正在尝试，但我从未做过这样的事情。 - Marco

我有另一个关于颜色的问题，但现在是使用按钮来为具有现有数据的表格上色。 - Marco