Matplotlib custom colors in scatter plot

Question

I would like to know if there is some elegant way to set discrete colors in a scatter plot based on the value of the entry. For a completly simplified example:

x=np.arange(100)
plt.scatter(x,x,c=x,cmap='jet')
plt.show()

So let's say I want all values lower than 40 green, those between 40 and 60 green, and those larger than 60 to be red. I know that I could create a loop with some if statements that generates a list based on those conditions, so that I have something like colorlist=['blue','blue',.....].

I know that this works, but is there some different way of doing it, or is this the easiest way?

blunova · Accepted Answer · 2022-06-22 10:01:59Z

4

You can do something like this:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.colors as mcol

cmap = mcol.ListedColormap(["blue","green","red"])

bounds = [0, 40, 60, 100]
norm = mcol.BoundaryNorm(bounds, cmap.N)

x=np.arange(100)
plt.scatter(x,x,c=x,cmap=cmap,norm=norm)
plt.show()

For further details, take a look here:

answered Jun 22, 2022 at 10:01

blunova

2,5804 gold badges13 silver badges27 bronze badges

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

Collectives™ on Stack Overflow

Matplotlib custom colors in scatter plot

1 Answer 1

Comments

Your Answer

Linked

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Linked

Related