Manipulating legend in Scatter plot in python plotly package

Question

I wanted to plot a scatter plot using plotly in python with specific marker colors. But when I was successful in doing that the legend did not correspond. All the labels are now of same color(blue) in legend. I want legend to represent the risks from my data with same colors as mentioned in code. Like this:

blue marker=moderately low
green marker=moderate
orange marker=moderately high

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

enter image description here

pass this paramter color_discrete_map = colorsIdx inside px.scatter() — vb_rises
– vb_rises, Commented Aug 6, 2019 at 12:41

nicolaskruchten · Accepted Answer · 2019-08-06 12:44:44Z

7

px.scatter() actually already knows how to do this:

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

answered Aug 6, 2019 at 12:44

nicolaskruchten

27.7k8 gold badges91 silver badges103 bronze badges

Sign up to request clarification or add additional context in comments.

1 Comment

nicolaskruchten Over a year ago

Great! feel free to upvote and/or accept the answer :)

Collectives™ on Stack Overflow

Manipulating legend in Scatter plot in python plotly package

1 Answer 1

1 Comment

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Linked

Related