KNN_2.py

import matplotlib.pyplot as plt 

points_1 = df[0:50].values.tolist()
points_2 = df[50:100].values.tolist()
points_3 = df[100:].values.tolist()

points_1 = np.array(points_1)
points_2 = np.array(points_2)
points_3 = np.array(points_3)

## Plot the data points to visualize 
plt.figure(figsize=(16,9))
plt.scatter(points_1[:,0],points_1[:,1],color='red',label='Setosa')
plt.scatter(points_2[:,0],points_2[:,1],color='black',label='Versicolor')
plt.scatter(points_3[:,0],points_3[:,1],color='green',label='Virginica')
plt.legend()
plt.show()