numpy阵列各轴上的K-平均值

我有一个形状为（N，M）的numpy数组a。我希望在每个M轴上运行K-mean（K=2）算法，并得到结果C，这是一个具有形状（K，M）的数组

---

M可以是一个非常大的数字。用python计算C的最快和最可扩展的方法是什么？

什么是

1-dk-mean（K=2）算法？

？嘿：）更新了这个问题。我所说的一维K-均值是指一维数据的K-均值算法。似乎形状

（M，）

意味着

K=1

@senderle你是对的。更新了问题。