In [21]:
# iscrtavanje skupa podataka zajedno sa novim elementima
plt.scatter(X[:,0], X[:,1])
plt.scatter(x1_test[0,0], x1_test[0,1], c=u'r')
plt.scatter(x2_test[0,0], x2_test[0,1], c=u'g')
plt.scatter(x3_test[0,0], x3_test[0,1], c=u'y');
# iscrtavanje skupa podataka zajedno sa novim elementima
plt.scatter(X[:,0], X[:,1])
plt.scatter(x1_test[0,0], x1_test[0,1], c=u'r')
plt.scatter(x2_test[0,0], x2_test[0,1], c=u'g')
plt.scatter(x3_test[0,0], x3_test[0,1], c=u'y');