Раскраска строк таблицы на основе значений столбцов

Question

Скажем, я выполняю машинное обучение, чтобы предсказать цвета, и я делаю таблицу с истинным цветом и предсказанным цветом. Я хочу покрасить всю строку в красный цвет, если прогнозируемый цвет и истинный цвет не совпадают. Например:

number = [1,2,3,4,5]
true_color = ['blue', 'red', 'green', 'yellow', 'blue']
prediction = ['blue', 'green', 'green', 'yellow', 'red']

fig = go.Figure(data=[go.Table(header=dict(values= ['Number', 
                                                   'True Color', 'Prediction'
                                                   ]),
                 cells=dict(values=[number, true_color ,
                                   prediction]))])

fig.show()

Как сделать 2-й и 5-й ряд красным.

Answer 1

Вы должны установить цвет строки, используя fill_color:

import plotly.graph_objects as go


number = [1, 2, 3, 4, 5]
true_color = ["blue", "red", "green", "yellow", "blue"]
prediction = ["blue", "green", "green", "yellow", "red"]


colors = ["red" if x != y else "white" for x, y in zip(true_color, prediction)]


fig = go.Figure(
    data=[
        go.Table(
            header=dict(values=["Number", "True Color", "Prediction"]),
            cells=dict(values=[number, true_color, prediction], fill_color=[colors]),
        )
    ]
)

fig.show()