Numpy Classification With Kmeans