/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

1 vote284 views3 answers
Opencv grayscale to rgb
Python
15 votes307 views2 answers
Loop throughthe key and the values of a dict in python
Python
3 votes281 views2 answers
Extract pdf text with python
Python
0 vote275 views1 answer
"not equal to" python symbol
Python
0 vote286 views1 answer
Python pick one item from list
Python
0 vote438 views1 answer
ImportError: cannot import name 'MigrateCommand' from 'flask_migrate'
Python Flask
9 votes279 views1 answer
Make tkinter btn disable
Python
0 vote270 views1 answer
Float division in python
Python
7 votes286 views1 answer
How to find where python is located
Python
3 votes289 views2 answers
Flask db migrate
Python