/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 vote106 views1 answer
How to read linux environment variable in python
Python
0 vote139 views1 answer
Load local data to django
Python Django
0 vote110 views1 answer
Find index corresponding to maximum value pandas
Python
6 votes128 views3 answers
Media url django
Python Django
0 vote115 views1 answer
Return all values in a list python
Python
1 vote102 views1 answer
Python añadir elementos a una lista
Python
7 votes111 views3 answers
Python selenium click element
Python
0 vote121 views1 answer
Get time format python2 hours minutes seconds milliseconds
Python
10 votes135 views2 answers
Adding static file and its usage in Django
Python Django
-1 vote131 views1 answer
Add widget in pyqt
Python