k-meansで、クラスタ番号とサンプルの名前を同時に返すメソッド

###k-meansで、クラスタ番号とサンプルの名前を同時に返すメソッド
pythonのk-meansで、クラスタ番号とそれに合致するサンプルの名前を、リストとして同時に返すメソッドはありますか？
例えば以下のようなデータがあった場合、[['0', '0'],['1', '0'],・・・・
というように左にサンプルの名前、右にクラスタ番号というふうにです。

一応、配列の結合は成功していて一次元配列を二次元配列に直しそれをconcatenateすることでできたのですが、あまりスマートではないな、と思い質問させていただきました。

どうかそのような便利なメソッド、またはもっとエレガントなやりかたがあれば教えていただきたいです・・・

一応公式サイトはみたのですが英語が並んでいて、あいまいな理解です・・・・

行動規範の内容に同意します

回答1件

ベストアンサー

そういったメソッドはありませんので、必要であればlabels_属性を見るかpredictメソッドでクラスタラベルを予測したのち、元のラベルと結合する質問文の方法で処理してください。

np.concatenateよりはnp.stackの方が綺麗かもしれません。

python
1>>> a = np.array([1,2,3])
2>>> b = np.array([4,5,6])
3>>> np.concatenate([a.reshape(-1, 1), b.reshape(-1, 1)], axis=1)
4array([[1, 4],
5       [2, 5],
6       [3, 6]])
7>>> np.stack([a, b], axis=1)
8array([[1, 4],
9       [2, 5],
10       [3, 6]])