/Python/How to plot labeled data with different colors

How to plot labeled data with different colors

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

x = [4,8,12,16,1,4,9,16]
y = [1,4,9,16,4,8,12,3]
label = [0,1,2,3,0,1,2,3]
colors = ['red','green','blue','purple']

fig = plt.figure(figsize=(8,8))
plt.scatter(x, y, c=label, cmap=matplotlib.colors.ListedColormap(colors))

cb = plt.colorbar()
loc = np.arange(0,max(label),max(label)/float(len(colors)))
cb.set_ticks(loc)
cb.set_ticklabels(colors)

Source: stackoverflow.com

Similar codes

0 vote145 views1 answer
How to get wikipedia page link in python
Python
1 vote134 views1 answer
F string curency format
Python
2 votes120 views2 answers
How to open a different version of python on my macc
Python
1 vote129 views1 answer
Ipython save session
Python
42 votes120 views6 answers
Python comment
Python
0 vote145 views1 answer
Debugging pytest in vscode
Python
0 vote133 views1 answer
Print treelib.Tree
Python
0 vote142 views5 answers
Representation of multidimensional array in data structure
Python
0 vote134 views1 answer
Pandas resamples stratified by columns values
Python
0 vote125 views1 answer
Var person
Python