df['target'] = kmean_model.predict(df) #plotting the first cluster plt.scatter(df[df['target']==1].iloc[:,0],df[df['target']==1].iloc[:,1],cmap = 'viridis')