k_means_1.py

import pandas as pd 
import numpy as np
from sklearn.utils import shuffle

## Load Iris dataset
df = pd.read_csv('/Users/rohith/Documents/Datasets/Iris_dataset/iris.csv') 
## Store the target vaue
classes = df['Species']  
## Drop the Id and Class values from dat
df = df.drop(['Id','Species'],axis=1) 
## Convert dataframe into list and then into a numpy array
data = df.values.tolist() 
data = np.array(data)
## Shuffle classes and data 
data,classes = shuffle(data,classes) 
## First 135 points are used for training and the rest is used for testing
train_data = data[:135]  
test_data = data[135:]