matplotlib - 无法无误地绘图

matplotlib - can't plot without errors

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

查看此附件的图片 post。

这是我的 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()

感谢您的每一个建议!

谢谢, 托马斯

绘制不需要的线条

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

(而且应该更快)