pyqt - 在TableView中更改row_cell颜色

时间:2013-03-25 18:26:07

标签: python qt pyqt4 tableview background-color

我有一个有三列的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),因为我已经在我的tableview中打开了一个现有的csv文件(使用按钮)。

1 个答案:

答案 0 :(得分:3)

您必须在模型中定义颜色,而不是在视图中定义:

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()