import numpy as np import matplotlib.pyplot as plt %matplotlib inline
X = np.array([[3,2],[2,4],[3,3],[1,4],[2,2],[-2,-3],[-3,-4],[-5,-2],[-1,-2],[-3,-3]])
y = np.array([1,1,1,1,1,2,2,2,2,2,])
plt.scatter(X[:,0],X[:,1],c=y);