Сохранение меток scikit kmeans в массиве со строкой для каждой метки

Question

Я делаю кластерный анализ с kmeans из sklearn. Теперь я хочу сохранить метки kmeans в массиве, и я хочу, чтобы каждая метка была в отдельной строке, но на данный момент я получаю их все в одной строке. Это мой код:

k1 = KMeans (n_clusters = 3) .fit (функции) pred = k1.predict (функции)

results = np.array ([k1.labels_])

results - это массив, в котором я сохраняю метки kmeans. Но на данный момент я получаю только один ряд со всеми метками. Но я хочу ряд для каждого лейбла. Заранее спасибо:)

Answer 1

Попробуйте это:

results=k1.labels_.reshape(k1.labels_.shape[0],1)

Это будет складываться в 1 столбец несколькими строками.