wxpythonでセルの色を変更する方法

Question

前提・実現したいこと

wxpythonを用いてGUIを作成しています．
データフレームを使ってCSVファイルを読み込んでからgridで表を作成します．
その時にもし同じ列に重複した要素があれば，セルの色を変更するというものを作りたいです．

発生している問題

表の作成まではできているのですが，セルの色がうまく変更できません

該当のソースコード

attr = wx.grid.GridCellAttr()
attr.SetBackgroundColour("#FFB6C1")

x=[]
for i in range(len(df)):
x.append([df.iat[i,j]]for i in range(len(df)))
    for j in range(len(df.columns)):
        x.append([df.iat[i,j]]for i in range(len(df)))
        grid.SetCellValue(i,j,df.iat[i,j])
        if df.iat[i,j] in x:
            grid.SetAttr(i,j,attr)
    x.clear()

試したこと

各列の要素を別のリストに保存し，そのリストに含まれていれば色を変更するというコードを書きましたが，実行しても色が変更されないです．

補足情報（FW/ツールのバージョンなど）

Python 3.7.4

Answer 1

こんな感じはいかがでしょうか？
wx.grid.GridTableBaseを継承したDataTable classに色を塗りたい部分の場所をx = [[1,1], [3,3], [5,5]]として渡してますー。
それで、GetAttrのところで色をつけてます！
もしよかったら見てみてくださいー。

import numpy as np
import pandas as pd

import wx
import wx.grid

EVEN_ROW_COLOUR = '#CCE6FF'
GRID_LINE_COLOUR = '#ccc'


class DataTable(wx.grid.GridTableBase):
    def __init__(self, data=None, hoge=[]):
        wx.grid.GridTableBase.__init__(self)
        self.headerRows = 1
        self.hoge = hoge
        if data is None:
            data = pd.DataFrame()
        self.data = data

    def GetNumberRows(self):
        return len(self.data)

    def GetNumberCols(self):
        return len(self.data.columns) + 1

    def GetValue(self, row, col):
        if col == 0:
            return self.data.index[row]
        return self.data.iloc[row, col - 1]

    def SetValue(self, row, col, value):
        self.data.iloc[row, col - 1] = value

    def GetColLabelValue(self, col):
        if col == 0:
            if self.data.index.name is None:
                return 'Index'
            else:
                return self.data.index.name
        return str(self.data.columns[col - 1])

    def GetTypeName(self, row, col):
        return wx.grid.GRID_VALUE_STRING

    def GetAttr(self, row, col, prop):
        attr = wx.grid.GridCellAttr()
        x = [row, col]
        if x in self.hoge:
            attr.SetBackgroundColour(EVEN_ROW_COLOUR)
        return attr


class MyFrame(wx.Frame):
    """
    Frame that holds all other widgets
    """

    def __init__(self):
        """Constructor"""
        wx.Frame.__init__(self, None, wx.ID_ANY, "Pandas")
        self._init_gui()
        self.Layout()
        self.Show()

    def _init_gui(self):
        df = pd.DataFrame(np.random.random((10, 5)))
        ##### 色を塗りたいところ
        x = [[1,1], [3,3], [5,5]]
        table = DataTable(df, x)

        grid = wx.grid.Grid(self, -1)
        grid.SetTable(table, takeOwnership=True)
        grid.AutoSizeColumns()

        sizer = wx.BoxSizer(wx.VERTICAL)
        sizer.Add(grid, 1, wx.EXPAND)
        self.SetSizer(sizer)

        self.Bind(wx.EVT_CLOSE, self.exit)

    def exit(self, event):
        self.Destroy()


if __name__ == "__main__":
    app = wx.App()
    frame = MyFrame()
    app.MainLoop()

Comments

[ponsuzunoko]

回答ありがとうございます
試して見ますね

Answer 2

以下のように処理することで色を変えることができました

attr = wx.grid.GridCellAttr()
attr.SetBackgroundColour("#FFB6C1")

x=[]

for j in range(len(df.columns)):
    for i in range(len(df)):
        #各行の要素を保存
        x =[df.iat[k,j] for k in range(len(df))]
        #それ自身を除く要素で考える
        x.remove(df.iat[i,j])
        grid.SetCellValue(i,j,df.iat[i,j])
        #他の行に同じ要素があれば色をつける
        if df.iat[i,j] in x:
            grid.SetAttr(i,j,attr)

    x.clear()