/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

6 votes200 views2 answers
Python code for where to save the figures
Python
0 vote272 views1 answer
Parallel loops in python
Python
9 votes172 views1 answer
Pandas sort by multiple columns
Python
0 vote264 views1 answer
Pandas earliest date in column
Python
0 vote191 views1 answer
Opening & creating hdf5 file
Python
0 vote209 views2 answers
Web scraping using python code
Python
5 votes195 views2 answers
Convert response to json python
Python
7 votes247 views3 answers
How to delete migrations in django
Python Django
0 vote188 views1 answer
Https://raw.githubusercontent.com/tim-yao/lighthouse-ci/d32f465bb6cda08ded4ce25c88c43a3103e4940a/.browserslistrc
Python
0 vote199 views1 answer
Django auth user model
Python