Kmeans Python

[wpdm_package id=’5662′]

[wpdm_package id=’5656′]

import pandas as pd

df = pd.read_csv("./autos_prepared.csv")

df.head()

%matplotlib inline

import matplotlib.pyplot as plt

plt.scatter(df["yearOfRegistration"], df["price"])
plt.show()

X = df[["yearOfRegistration", "price"]]

from sklearn.preprocessing import StandardScaler

scaler = StandardScaler()
X_transformed = scaler.fit_transform(X)

from sklearn.cluster import KMeans

model = KMeans(n_clusters = 3)
model.fit(X_transformed)

labels = model.labels_

%matplotlib inline

import matplotlib.pyplot as plt

plt.scatter(df["yearOfRegistration"], df["price"], c = labels)
plt.show()