/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

9 votes204 views1 answer
Area of triangle
Python
0 vote254 views1 answer
Spacy nlp load
Python
0 vote189 views1 answer
Black code formatter
Python
0 vote231 views1 answer
Add header info in django response
Python Django
9 votes177 views2 answers
Python execute string
Python
5 votes205 views2 answers
Scroll to element python selenium
Python
0 vote182 views1 answer
Intersection of 3 array in O(n) python
Python
0 vote191 views1 answer
Time counter in python
Python
0 vote176 views1 answer
Python min function time complexity
Python
9 votes186 views2 answers
Get a list of all files python
Python