使用matplotlib可视化聚类结果

Question

我在一个大小为（11464,2622）的数据集上运行了一个模糊的c-means聚类算法，该数据集有10个聚类，并产生一个（10,11464）大小的中心，其最终模糊c-分区矩阵的大小（10,2622）

我的挑战是如何为我的数据集中的每个数据点绘制已分配的群集。我在下面尝试了这个但是没有用。

cluster_membership = np.argmax(u, axis=0)
colors = ['b', 'orange', 'g', 'r', 'c', 'm', 'y', 'k', 'Brown','ForestGreen']

for j in range(10):
  x = data[:, 0]
  y = data[:, 1]
  plt.scatter(x[cluster_membership == j], y[cluster_membership == j], '.', color=colors[j])

其中u是分区矩阵。但我收到ufunc 'sqrt' not supported for the input types, and the inputs could not be safely coerced to any supported types according to the casting rule ''safe''

以下的错误

有人能指出我正确的方向来获取我的可视化效果吗？此致

P.S我正在关注这个例子here