在Numpy中對多維數組中的點集進行矢量化查找中心

Question

我有一個多維數組，它有 100 萬組 3 個點，每個點都是由 x 和 y 指定的坐標。 調用這個數組pointVec，我的意思是

np.shape(pointVec) = (1000000,3,2)

我想找到每組 3 個點的中心。 一種明顯的方法是遍歷所有 100 萬個集合，在每次迭代中找到每個集合的中心。 但是，我聽說矢量化是 Numpy 的強項，所以我正在嘗試使其適應這個問題。 由於這個問題非常適合迭代，因此我不了解如何使用矢量化來解決這個問題，或者使用矢量化是否有用。

Answer 1

這取決於您如何定義三點的中心。 但是，如果是平均坐標，如評論中提到的@Quang，您可以在numpy中沿特定軸取平均值：

pointVec.mean(1)

這將取沿軸 = 1（它是具有 3 個點的第二個軸）的平均值並返回一個 (1000000,2) 形狀的數組。