└── cluster.py


/cluster.py:
--------------------------------------------------------------------------------
 1 | import numpy as np
 2 | import pandas as pd
 3 | import matplotlib.pyplot as plt
 4 | from sklearn.cluster import KMeans
 5 | data = pd.read_csv("/content/sample_data/partywise.csv")
 6 | plt.scatter(data["Party"], data["Won"])
 7 | plt.xlim(-180,180)
 8 | x = data.iloc[:,1:3]
 9 | data.dtypes
10 | kmeans = KMeans(3)
11 | kmeans.fit(x)
12 | identified_clusters = kmeans.fit_predict(x)
13 | identified_clusters
14 | data_with_clusters = data.copy()
15 | data_with_clusters['Clusters'] = identified_clusters
16 | plt.scatter(data_with_clusters['Won'],data_with_clusters['Party'],c=data_with_clusters['Won'],cmap='rainbow')


--------------------------------------------------------------------------------