在pythonplotly包中操作散点图中的图例

Vin*_*ari 2 python scatter-plot plotly jupyter-notebook

我想使用 python 中的plotly 和特定标记颜色来绘制散点图。但当我成功做到这一点时,图例并不对应。现在,图例中的所有标签都具有相同的颜色(蓝色)。我希望图例用与代码中提到的相同颜色来表示数据的风险。像这样:

蓝色标记=中等低
绿色标记=中等
橙色标记=中等高

import plotly.express as px
import plotly.graph_objects as go

colorsIdx = {'Moderately Low': 'blue', 'Moderate': 'green', 'Moderately High': 'orange'}
cols = data['Risk'].map(colorsIdx)

fig = px.scatter(data, x="1_Yr_Return", y="Expense_Ratio", color='Risk')

fig.update_traces(marker=dict(size=12, color=cols), selector=dict(mode='markers'))
fig.show()
Run Code Online (Sandbox Code Playgroud)

在此输入图像描述

nic*_*ten 7

px.scatter()实际上已经知道如何做到这一点:

import plotly.express as px

colorsIdx = {'Moderately Low': 'blue', 'Moderate': 'green', 'Moderately High': 'orange'}

fig = px.scatter(data, x="1_Yr_Return", y="Expense_Ratio", 
                 color='Risk', color_discrete_map=colorsIdx)

fig.show()
Run Code Online (Sandbox Code Playgroud)