Визуализация результата кластеризации с помощью matplotlib

Question

Я запустил нечеткий алгоритм кластеризации c-средних для набора данных размером (11464, 2622) с 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''

Может ли кто-нибудь указать мне правильное направление, чтобы получить мою визуализацию? С уважением

PS Я следую этому примеру здесь