Pyqt改变表6中一个单词的颜色

Question

我需要帮助更改QTableView中一个单词的颜色。在

我需要做的是：当我打开表格中的csv文件时，现在我的columns~行中有句子/单词，我需要给一些单词涂颜色，比如：

在这里你可以看到我的程序与我的csv。 http://i.imgur.com/uIBQ2ch.jpg

示例：在[3][0]（4行，1列）中，我有一个单词“filme”，正如您在我的图像中看到的那样。在

我想给这个词上色，如果存在同一个词，在[3][2]（4行，3列）中，我也要给这个词上色

以下是我的完整代码：

#!/usr/bin/env python
#-*- coding:utf-8 -*-

import sip
sip.setapi('QVariant', 2)
import sys
reload(sys)
sys.setdefaultencoding("utf-8")
from PyQt4.QtCore import *
from PyQt4.QtGui import *
import csv


try:
    _fromUtf8 = QString.fromUtf8
except AttributeError:
    _fromUtf8 = lambda s:s



class MyWindow(QWidget):
def __init__(self, fileName, parent=None):
    super(MyWindow, self).__init__(parent)
    self.fileName = fileName

    self.model = QStandardItemModel(self)

    self.tableView = QTableView(self)
    self.tableView.setModel(self.model)
    self.tableView.horizontalHeader().setStretchLastSection(True)

    self.pushButtonLoad = QPushButton(self)
    self.pushButtonLoad.setText("Load Csv File!")
    self.pushButtonLoad.clicked.connect(self.on_pushButtonLoad_clicked)

    self.pushButtonWrite = QPushButton(self)
    self.pushButtonWrite.setText("Write Csv File!")
    self.pushButtonWrite.setStyleSheet('color:red;background-color:rgb(155, 255, 153);border:1px solid purple;')
    self.pushButtonWrite.clicked.connect(self.on_pushButtonWrite_clicked)

    self.layoutVertical = QVBoxLayout(self)
    self.layoutVertical.addWidget(self.tableView)
    self.layoutVertical.addWidget(self.pushButtonLoad)
    self.layoutVertical.addWidget(self.pushButtonWrite)


def loadCsv(self, fileName):
    with open(fileName, "rb") as fileInput:
        for row in csv.reader(fileInput):    
            items = [
                QStandardItem(field.decode('utf8'))
                for field in row
            ]
            self.model.appendRow(items)
            self.tableView.resizeRowsToContents()
            self.model.setHeaderData(0, Qt.Horizontal, "Feature")
            self.model.setHeaderData(1, Qt.Horizontal, "Polarity(-1,0,1)")
            self.model.setHeaderData(2, Qt.Horizontal, "texto")
            self.tableView.setColumnWidth(2,1000)
            self.tableView.resizeRowsToContents() 

def writeCsv(self, fileName):
    with open(fileName, "w") as fileOutput:
        writer = csv.writer(fileOutput)
        for rowNumber in range(self.model.rowCount()):
            fields = [
                self.model.data(
                    self.model.index(rowNumber, columnNumber),
                    Qt.DisplayRole
                )
                for columnNumber in range(self.model.columnCount())
            ]
            writer.writerow(fields)
@pyqtSlot()
def on_pushButtonWrite_clicked(self):
    self.writeCsv(self.fileName)

@pyqtSlot()
def on_pushButtonLoad_clicked(self):
    self.loadCsv(self.fileName)

if __name__ == "__main__":
import sys

app = QApplication(sys.argv)
app.setApplicationName('MyWindow')

main = MyWindow("marcoteste.csv")  

main.show()

sys.exit(app.exec_())

Answer 1

据我所知，实际上有两种方法

第一个就像Oleg说的，使用委托，但是我的实现有点愚蠢，这里是pesudo代码

self.table = QTableView()
self.model = QStandItemModel()
self.table.setMode(model)
self.table.pressed.connect(self.detectWord)

def detectWord(self):
    self.to_color = []
    for row in range(self.model.rowCount()):
        for column in range(self.model.colunCount()):
            raw = self.model.data(self.model.index(row, column), Qt.DisplayRole)
            # needs decode, it's QString/utf-16 by default
            if 'word' in codecs.decode(raw.toString(), 'utf-16'):
                self.to_color.append([row, column])

    # apply delegate
    self.table.setItemDelegate(ColorColumn(self.to_color, self)

class ColorColumn(QItemDelegate):
    def __init__(self, to_color, parent = None):
        self.to_color = to_color

    def paint(self, painter, option, index):
        masked = False
        for pos in self.to_delete:
            if index.row() == pos[0] and index.column == pos[1]:
                text = index.model().data(index).toString()
                palette = QApplication.palette()

                document = QTextDocument()
                document.setDefaultFont(option.font)
                document.setPlainText(text)
                painter.save()
                palette.fillRect(option.rect, QColor(255, 255, 0, 50))
                document.drawContent(painter)
                painter.restore()
                masked = True

        if not masked:
            QItemDelegate.paint(self, paint, option, index)

但这会以某种方式一遍又一遍地扫描整个数据，有潜在的性能问题

第二个更标准，使用QAbstractTableModel，参见shuman's answer

基本上做以下事情

1. 创建一个类来表示一个数据项（继承对象）

   此外，def contain（self，column，to_color_word）表示是否应该对其着色

2. 创建一个表示表的模型（继承QAbstractTableModel）

   和他一样，在数据（self，index，role）中，也就是“read”或model，在backgroundrole中，使用您在上面定义的方法来表示颜色

3. 要在顶层组合的setModel

这比使用QstandardItem更合适