/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 vote187 views1 answer
How to take index first and last in python in one way
Python
1 vote211 views1 answer
Deleting DataFrame row in Pandas based on column value
Python
15 votes190 views3 answers
Python - gropuby based on 2 variabels
Python
0 vote194 views1 answer
How to sort a list in python with if
Python
4 votes205 views2 answers
Delete rows in dataframe pandas
Python
0 vote233 views1 answer
Series object has no attribute columns
Python
0 vote213 views2 answers
How to import image in python
Python
35 votes196 views6 answers
Pandas merge dataframes
Python
1 vote210 views1 answer
Pandas read csv read all rows except one
Python
5 votes204 views1 answer
Sphere volume formula
Python