Marker Color based on String Value Plotly Python

Question

Pretty simple question here, but I cannot seem to figure it out. I have a dataframe column with 3 possible values:

print(df['Site'])

0        ABC
1        EFG
2        ABC
3        ABC
4        HIJ

When I plot, each site has the same color, but I would each marker to have a color to represent the site. I have tried to following 2 options with no luck:

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

I know that Plotly/Dash wants an integer. Is there a work around for this? How can I plot site color in the markers?

alec_djinn · Accepted Answer · 2020-06-02 15:06:37Z

3

I would start with a dictionary mapping colors and sites.

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

You can use any HTML color name there.

Then, you can make a list of colors based on your dict.

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

Finally, you can pass marker_color=colors to fig.add_trace()

edited Jun 2, 2020 at 15:06

answered Jun 2, 2020 at 15:01

alec_djinn

10.9k9 gold badges57 silver badges77 bronze badges

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

Comments

nicolas.f.g · Accepted Answer · 2022-04-02 10:18:18Z

0

If you use Plotly Express, you can pass the parameter color_discrete_map in px.scatter(). For example:

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

Source: Manipulating legend in Scatter plot in python plotly package

answered Apr 2, 2022 at 10:18

nicolas.f.g

1,26111 silver badges14 bronze badges

Collectives™ on Stack Overflow

Marker Color based on String Value Plotly Python

2 Answers 2

Comments

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related