In [5]:
x_test_1 = np.array ([[2,1]])
In [6]:
x_test_1.shape
Out[6]:
(1, 2)
In [7]:
plt.scatter(X[:,0],X[:,1],c=y)
plt.scatter(x_test_1[0,0],x_test_1[0,1],c='r')
# s obzirom na to da računamo razdaljinu između tački, dobra je praksa da se ose podese na 'equal' kako bi  
# iste dužine odgovarale i na x i na y osi. 
plt.axis('equal');