如何在 Python 中更改 plotly.express 的颜色和大小？

Question

I'm using plotly.express to make a plot of my Artificial Neural Networks in Python. I'll show my code:

import plotly.express as px
fig = px.scatter(newDf, x="Glucose", y="Age", color="Outcome")
fig.show()

I would like the sidebar of "color" (comprehends the Outcome) has only 0 and 1 values (instead of 0, 0.2, 0.4...). Plus, I would like to change the color of the points in the scatter plot. Currently, the plot has blue(0) and yellow(1) points and, expecially the yellow ones, are difficult to see.

Answer 1

Presuming your outcome only has 0 and 1 values, the easy way to make plotly treat it as a discrete value would be to convert it to a string.

df["Outcome"] = df["Outcome"].astype(str) #convert to string

# you can convert back to numeric later, if needed
df["Outcome"] = df["Outcome"].astype(float) 

Now, to change the colors you can use the argument color_discrete_sequence like this:

mport plotly.express as px
fig = px.scatter(newDf, x="Glucose", y="Age", color="Outcome", 
                 color_discrete_sequence= ["blue", "red"])
fig.show()