How to assign color value using plotly based on column value?

Question

I am trying to plot k-means cluster using plotly, but i am having trouble in assigning colors based on there groups? I have following data frame.

group: cluster number

I am using this for scatter plot in plotly.

clustered.iplot(kind='scatter',x='value1',y='value2', colors = {'[clustered['group']==1]':'green', '[clustered['group']==0]':'yellow'},mode='markers',size=10)

Its wrong because it will only get True and false for color dict object. How i can can associate these group values so that color of points appear differently in the plot.

nicolaskruchten · Accepted Answer · 2019-07-14 02:00:43Z

3

Instead of using cufflinks you can use the new Plotly Express library (https://plotly.express) to do this with:

px.scatter(clustered, x='value1', y='value2', color='group')

answered Jul 14, 2019 at 2:00

nicolaskruchten

27.7k8 gold badges91 silver badges103 bronze badges

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

Comments

zhavorsa · Accepted Answer · 2019-07-12 12:49:27Z

2

If you only have two clusters, you can map the values:

clustered.iplot(kind='scatter', x='value1', y='value2', colors=clustered['group'].map({0:'yellow', 1:'green'}), mode='markers',size=10)

answered Jul 12, 2019 at 12:49

zhavorsa

4042 silver badges10 bronze badges

1 Comment

M_S_N Over a year ago

It didn't worked. i am getting this error ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Collectives™ on Stack Overflow

How to assign color value using plotly based on column value?

2 Answers 2

Comments

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

2 Answers 2

Comments

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related