基于字符串值的标记颜色 Plotly Python

Question

这里的问题很简单，但我似乎无法弄清楚。 我有一个包含 3 个可能值的 dataframe 列：

print(df['Site'])

0        ABC
1        EFG
2        ABC
3        ABC
4        HIJ

当我 plot 时，每个站点都有相同的颜色，但我会为每个标记都用一种颜色来代表该站点。 我尝试过以下 2 个选项，但没有成功：

fig.add_trace(go.Scatter(x=df['Reported Date'], y=df['Score'],
    mode='markers', 
    marker= dict(color=list(map(SetColor, df['Site']))),
    name='Points', 
    text="Site: " + df['Site'].map(str) + " " + "Ticket: "+ df['Ticket ID'].map(str)))

fig.add_trace(go.Scatter(x=df['Reported Date'], y=df['Score'],
    mode='markers', 
    marker_color= df['Site'],
    name='Points', 
    text="Site: " + df['Site'].map(str) + " " + "Ticket: "+ df['Ticket ID'].map(str)))

我知道 Plotly/Dash 想要一个 integer。有解决办法吗？ 我怎样才能在 plot 站点颜色中标记？

Answer 1

我将从映射 colors 和站点的字典开始。

d = {
    'ABC':'red',
    'DEF':'green',
    'GHI':'black'
}

您可以在那里使用任何HTML 颜色名称。

然后，您可以根据您的 dict 列出 colors 列表。

colors = [d[k] for k in df['Site'].values]

最后，您可以将marker_color=colors传递给fig.add_trace()

Answer 2

如果使用 Plotly Express，可以在px.scatter()中传入参数color_discrete_map 。 例如：

px.scatter(df, x='datetime', y='measurement', 
           color='target', 
           color_discrete_map={'value1': 'green', 'value2': 'red'})

资料来源： 在Scatter plot中操纵图例 python plotly package