使某些点在 Plotly Express Scatterplot 不同 Colors

Question

目前，我有一个散点图，每个点代表一位导演，基于他们的制作预算和利润。 我想突出其中一些以自己独特的颜色脱颖而出的点，并创建一个图例来说明这些点中的每一个是谁。 例如，我想突出显示 Joe Russo，方法是在图表上将他的点设为绿色，然后将他添加到图例中。

现在，我有很好的散点图。 这是我当前的代码：

BudProfDirector = px.scatter(BudProfDirectorDF, x = 'production_budget', y = 'Worldwide_Profit', hover_name='primary_name')

这给出了这张图： Plotly 散点图

我还有一份我想强调的董事名单：

DirectorsToHighlight = ['Pierre Coffin', 'Chris Renaud', 'Kyle Balda','Zack Snyder', 'Christopher Nolan', 'Joe Russo']

但不知道从哪里到 go

Answer 1

要为特定点着色并将其反映在图例中，在这种情况下，最简单的方法是通过在特定点创建具有相同点大小的新数据框来添加图形。 该图是根据添加的数据创建的。

import plotly.express as px

DirectorsToHighlight = ['Pierre Coffin', 'Chris Renaud', 'Kyle Balda', 'Christopher Nolan', 'Joe Russo']

BudProfDirector = px.scatter(BudProfDirectorDF,
                             x='production_budget',
                             y='Worldwide_Profit',
                             hover_name='primary_name')
df = BudProfDirectorDF[BudProfDirectorDF['primary_name'].isin(DirectorsToHighlight)]

for row in df.itertuples():
    print(row)
    BudProfDirector.add_scatter(x=[row.production_budget],
                                y=[row.Worldwide_Profit],
                                hovertext=[row.primary_name],
                                hovertemplate='<b>%{hovertext}</b><br><br>production_budget=%{x}<br>Worldwide_Profit=%{y}<extra></extra>',
                                name=row.primary_name,
                                mode='markers')

BudProfDirector.show()