如何仅在TableWidget的列设置backgroundColor？

from PyQt5.QtWidgets import QApplication, QTableWidget, QTableWidgetItem, QStyledItemDelegate
from PyQt5.QtGui import QColor, QPainter

class ColumnColorDelegate(QStyledItemDelegate):
    def paint(self, painter, option, index):
        # 获取当前单元格的行和列
        row = index.row()
        column = index.column()
        
        # 设置第2列的背景颜色为红色
        if column == 1:
            painter.fillRect(option.rect, QColor(255, 0, 0))
        
        # 绘制文本
        self.drawDisplay(painter, option, option.rect, index.data())

app = QApplication([])
tableWidget = QTableWidget(4, 3)
tableWidget.setItemDelegateForColumn(1, ColumnColorDelegate())

# 添加数据
tableWidget.setItem(0, 0, QTableWidgetItem("A"))
tableWidget.setItem(0, 1, QTableWidgetItem("B"))
tableWidget.setItem(0, 2, QTableWidgetItem("C"))
tableWidget.setItem(1, 0, QTableWidgetItem("D"))
tableWidget.setItem(1, 1, QTableWidgetItem("E"))
tableWidget.setItem(1, 2, QTableWidgetItem("F"))
tableWidget.setItem(2, 0, QTableWidgetItem("G"))
tableWidget.setItem(2, 1, QTableWidgetItem("H"))
tableWidget.setItem(2, 2, QTableWidgetItem("I"))
tableWidget.setItem(3, 0, QTableWidgetItem("J"))
tableWidget.setItem(3, 1, QTableWidgetItem("K"))
tableWidget.setItem(3, 2, QTableWidgetItem("L"))

tableWidget.show()
app.exec_()