python - 根据列值对表行进行着色

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