import numpy as np
import numpy.random as rnd
import matplotlib.pyplot as plt

# génération des points de la classe 1 & 2
N=100
x = np.vstack((rnd.randn(N,2)+2,rnd.randn(N,2)-2)) # données 2D
y = np.ones(2*N) # étiquettes
y[:N] = -1

# comment afficher chaque classe d'une couleur???

# solution 1
ind1 = np.where(y==1) 
ind2 = np.where(y==-1)
plt.figure()
plt.plot(x[ind1, 0],x[ind1, 1], 'b+') # aff en croix bleues
plt.plot(x[ind2, 0],x[ind2, 1], 'r*') # aff en étoiles rouges