matplotlib - 可以无错误地绘图

Question

我是matplotlib的新手，我正在学习如何使用它。使用我的代码，我试图绘制多个散点图，但总有不需要的线条，我无法摆脱它。

见本文附带的图片。

这是我的Python代码：

import matplotlib.pyplot as plt
from sklearn import datasets
from sklearn import cluster

iris = datasets.load_iris()

for i in range(1,7):
    kmeans = cluster.KMeans(n_clusters=i).fit(iris.data)
    Centers = kmeans.cluster_centers_
    plot_nr = 230+i
    plt.subplot(plot_nr)
    text = str(i) + ' Cluster'
    plt.scatter(iris.data[:,0], iris.data[:,1], c=kmeans.labels_)
    plt.scatter(Centers[:,0], Centers[:,1], c='r', s=150, marker='*')
    plt.title(text)

plt.savefig('Iris.png')
plt.show()

我感谢你的每一个建议！

谢谢， 托马斯

绘制不需要的行

Answer 1

修改第一行plt.subplots(231, figsize=(18,12))并将231替换为2, 3, 231在两行之间创建一个单元格。可能重复了6次

（它应该更快）