kmeans聚类中的簇数错误

Question

我只能有5个集群。我想要有6个集群。该代码给出了以下错误：

IndexError: list index out of range

以下一行：

plt.plot(X[i][0], X[i][1], colors[labels[i]], markersize = 10)

我不知道发生了什么。请帮忙。

Answer 1

确保您的colors数组足够大。

此外，学习阅读错误消息和跟踪并执行基本调试任务以了解您的错误。

显然，这里的错误是在绘图中，而不是在kmeans中。您可以检查找到多少个kmeans，这可能更容易找到。