python 简单计算质心的方法

最新推荐文章于 2024-07-28 03:06:31 发布

原创最新推荐文章于 2024-07-28 03:06:31 发布 · 5.7k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#python #numpy #质心

python3 专栏收录该内容

39 篇文章

订阅专栏

这段代码演示了如何使用numpy库中的mean函数沿着指定轴计算矩阵的平均值。输入矩阵a是一个二维数组，输出b是矩阵a按列的平均值。

from scipy import ndimage
import numpy as np

a= np.array([
    [0,  0,  0,  0],
    [0,  1,  0,  0],
    [0,  2,  0,  0],
    [1,  0,  0,  0],
    [1,  1,  0,  1],
    [1,  2,  0,  1],
    [2,  0,  0,  0],
    [2,  1,  0,  0],
    [2,  2,  0,  0]
])

b = np.mean(a,axis=0)

print(b)

'''
[1.         1.         0.         0.22222222]
'''