/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

68 votes224 views6 answers
Python open file
Python
12 votes194 views1 answer
Convert integer column to string in pandas
Python
29 votes230 views1 answer
Sort a dataframe by a column valuepython
Python
33 votes239 views5 answers
Python check if string contains symbols
Python
1 vote237 views1 answer
Two elements at a time in list comprehension
Python
0 vote236 views1 answer
Django reverse accessor clashes for abstract class
Python
-1 vote231 views1 answer
How to make take command in python
Python
3 votes223 views2 answers
How to check if text is in upper case in python
Python
1 vote228 views1 answer
Set pop in python
Python
2 votes235 views1 answer
How to make exe from.py file
Python