数组中多行上的 np.linalg.norm

Question

我试图在一个看起来像这样的数组中获取每个向量的大小；

[[0.1,1.0,2.0]
[0.5,2.0,1.0]
[0.4,3.0,0.5]]

我想把这些放到一个像这样的新 numpy 数组中；

[[1.2]
[2.0]
[2.1]]

目前我正在用一个循环来做这个；

for i in range(x)):
    d = vec2[i] - vec1[i]
    dist[i] = np.linalg.norm(d)

但是我想用纯 numpy 来做这件事（上面是 numpy 数组）有没有办法在一行上做到这一点？ 这不起作用，因为我得到的值不相同；

d = vec2 - vec1
dist = np.linalg.norm(d, axis=1)

Answer 1

使用axis = 0或axis = 1，这应该对每个数组执行计算，具体取决于您前进的方向