/Python/How to plot two different class in different colour in python

How to plot two different class in different colour in python

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

How to plot two different class in different colour in python

groups = data.groupby("Category")
for name, group in groups:
    plt.plot(group["X Value"], group["Y Value"], marker="o", linestyle="", label=name)
plt.legend()

Similar codes

1 vote208 views1 answer
How to use enumerate instead of range and len
Python
9 votes196 views4 answers
Scikit learn random forest classifier
Python
1 vote234 views1 answer
Url in form action django
Python Django
9 votes189 views2 answers
Flask flash
Python
0 vote193 views1 answer
Unit converter idle python
Python
0 vote174 views1 answer
TypeError: 'frozenset' object is not callable
Python
1 vote167 views1 answer
Matrix power python
Python
14 votes183 views2 answers
How to read a website in python
Python
2 votes179 views3 answers
Numpy random
Python
0 vote224 views1 answer
Python three periods
Python