K-Means聚类可视化

def solve():
    import pandas as pd
    import matplotlib.pyplot as plt
    from sklearn.cluster import KMeans
    df = pd.read_csv("https://data.zuihe.com/clustering.csv")
    X = df[["feature1","feature2"]]
    km = KMeans(n_clusters=3, random_state=42, n_init=10).fit(X)
    plt.scatter(X["feature1"], X["feature2"], c=km.labels_, cmap="viridis", alpha=0.7)
    plt.scatter(km.cluster_centers_[:,0], km.cluster_centers_[:,1], c="red", marker="X", s=200, label="Centers")
    plt.legend(); plt.title("K-Means Clustering"); plt.show()
    print(f"{km.inertia_:.2f}")