/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

2 votes198 views1 answer
How to split image dataset into training and test set keras
Python
8 votes183 views3 answers
F.readlines python
Python
0 vote240 views1 answer
Kivy video recorder
Python
0 vote184 views1 answer
C4d python ReferenceError: could not find 'main' in tag 'Null'
Python
0 vote198 views1 answer
Matplotlib matrix plot
Python
0 vote181 views1 answer
User passes test django
Python
1 vote180 views1 answer
Python dynamic object
Python
0 vote190 views1 answer
Dynamic array python numpy
Python
0 vote185 views1 answer
Get maximum value index after groupby
Python
0 vote185 views2 answers
Change index of dataframe with list
Python