如何在python中高效计算,两个数组中每对点的加权点平均值(质心)


# Table with weight 2/3
a = np.array(
[[0, 0],
[12, 12]]
)
# Table with weight 1/3
b = np.array(
[[12, 6],
[9, 3]]
)
# Returned table
c = np.array(
[[4, 2],
[11, 9]]
)

我有a,b(每个都有一些点(,我想在给定它们的权重的情况下有效地计算,矩阵c保持成对的点平均值。有点像他们的重心。

我该怎么做?感谢

您可以使用简单的基本numpy矩阵运算:

c = a * weight_a + b * weight_b
# With your example : 
c = a * 2 / 3 + b * 1 / 3
# array([[ 4.,  2.],
#       [11.,  9.]])

最新更新