我有一個多維數組，其中有 100 萬組 3 個點，每個點都是由 x 和 y 指定的坐標。調用這個數組pointVec，我的意思是np.shape(pointVec) = (1000000,3,2)我想找到這組 3 個點中每個點的中心。一種明顯的方法是迭代所有 100 萬個集合，在每次迭代時找到每個集合的中心。然而，我聽說矢量化是 Numpy 的強項，所以我正在嘗試使其適應這個問題。由于這個問題非常直觀地適合迭代，因此我不知道如何使用矢量化來解決這個問題，或者使用矢量化是否有用。
1 回答
蕭十郎
TA貢獻1815條經驗 獲得超13個贊
這取決于您如何定義三點中心。但是，如果是平均坐標，就像評論中提到的@Quang，您可以沿numpy中的特定軸取平均值：
pointVec.mean(1)
這將沿 axis=1（這是具有 3 個點的第二個軸）取平均值并返回 (1000000,2) 形狀的數組。
添加回答
舉報
0/150
提交
取消