/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 vote341 views1 answer
Increase colorbar ticksize
Python
0 vote308 views1 answer
Bytestring python
Python
5 votes327 views1 answer
How to use a string variable as a variable name in python
Python
0 vote280 views10 answers
Introduction to distutils in python
Python
0 vote301 views1 answer
Pydrive list shared folder
Python
0 vote330 views1 answer
How to copy list python
Python
0 vote347 views1 answer
Python subprocess stdout to dev null
Python
28 votes346 views3 answers
Datetime one month ago python
Python
0 vote352 views1 answer
Python class udp
Python
0 vote352 views1 answer
Install matplotlib on nvidia jetson nx
Python