數組中多行上的 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，這應該對每個數組執行計算，具體取決於您前進的方向