/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

5 votes356 views1 answer
Get index in foreach py
Python
2 votes337 views1 answer
How to move the pointer on screen using python
Python
4 votes397 views6 answers
Eval function in python
Python
4 votes336 views1 answer
Django objects.create()
Python Django
1 vote339 views1 answer
Python ndarray string array into int
Python
0 vote326 views1 answer
Mnist
Python
46 votes332 views3 answers
Pandas apply function to every row
Python
11 votes313 views2 answers
Cross validation python
Python
1 vote283 views1 answer
Dataframe sort values descending
Python
0 vote288 views1 answer
Atan of number python
Python