在 1 张图像中绘制散点图

Question

我正在尝试使用散点图绘制我的数据，但是我似乎无法将 3 个图放入 1 个图像中我该如何解决这个问题：

nu_cluster = 3
kmeans  = KMeans(n_clusters=nu_cluster,random_state=0)
data_df["cluster"] = kmeans.fit_predict(X_std)
print("after Kmeans predict")# visualization
plt.figure(figsize=(8, 6))
for i in range(nu_cluster):
    cluster_data = data_df[data_df["cluster"] == i]#return a boolean and then passed to data_df
    plt.scatter(cluster_data["charges"], cluster_data["age"],c=[plt.cm.viridis(i / (nu_cluster - 1))] ,label=f"Cluster {i + 1}")
    plt.xlabel("Charges")
    plt.ylabel("Age")
    plt.title("Cluster of age against charges", fontsize=16, fontweight="bold")
    plt.legend(loc="lower right")
    plt.show()

我尝试使用matplotlib提供的figure()函数，但没有成功。

plt.figure(figsize=(8,7))

Answer 1

将

plt.show()

移到循环之外。一旦您调用

plt.show()

，该图形将被“丢弃”，后续绘图命令会自动创建一个新图形。