/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

3 votes162 views2 answers
How to download youtube playlist using python
Python
3 votes237 views3 answers
Flask migrate
Python Flask
4 votes166 views4 answers
Count unique elements in list python
Python
0 vote135 views1 answer
Check tensor type tensorflow
Python
0 vote134 views1 answer
Display multiple dataframe as table jupyter notebook
Python
0 vote147 views1 answer
How to know python bit version
Python
10 votes133 views3 answers
Pygame.key.get_pressed()
Python
-1 vote137 views1 answer
How to print memory address in python
Python
3 votes235 views1 answer
Mouse bottom in pygame
Python
5 votes145 views2 answers
Pygame flip image
Python