2

说我有一个像这样的 (3,3,3) 数组。 

array([[[1, 1, 1],
        [1, 1, 1],
        [0, 0, 0]],

       [[2, 2, 2],
        [2, 2, 2],
        [2, 2, 2]],

       [[3, 3, 3],
        [3, 3, 3],
        [1, 1, 1]]])

如何获得对应于 3 个值的每个向量与第零个值之间的欧几里得距离的 9 个值？

比如做numpy.linalg.norm([1,1,1] - [1,1,1])2 次，然后做norm([0,0,0] - [0,0,0])，然后norm([2,2,2] - [1,1,1])2 次，，，norm([2,2,2] - [0,0,0])然后norm([3,3,3] - [1,1,1])2 次，最后norm([1,1,1] - [0,0,0])。

有什么好的方法可以矢量化吗？我想将距离存储在 (3,3,1) 矩阵中。

结果将是：

array([[[0. ],
        [0. ],
        [0. ]],

       [[1.73],
        [1.73],
        [3.46]]

       [[3.46],
        [3.46],
        [1.73]]])
4

3 回答 3

2

keepdims参数是在 numpy 1.7 中添加的，您可以使用它来保持总和轴：

np.sum((x - [1, 1, 1])**2, axis=-1, keepdims=True)**0.5

结果是：

[[[ 0.        ]
  [ 0.        ]
  [ 0.        ]]

 [[ 1.73205081]
  [ 1.73205081]
  [ 1.73205081]]

 [[ 3.46410162]
  [ 3.46410162]
  [ 0.        ]]]

编辑

np.sum((x - x[0])**2, axis=-1, keepdims=True)**0.5

结果是：

array([[[ 0.        ],
        [ 0.        ],
        [ 0.        ]],

       [[ 1.73205081],
        [ 1.73205081],
        [ 3.46410162]],

       [[ 3.46410162],
        [ 3.46410162],
        [ 1.73205081]]])
于 2013-05-10T07:35:42.840 回答
1

您可能需要考虑scipy.spatial.distance.cdist()，它可以有效地计算两个输入集合中的点对之间的距离（使用标准欧几里得度量等）。这是示例代码：

import numpy as np
import scipy.spatial.distance as dist

i = np.array([[[1, 1, 1],
               [1, 1, 1],
               [0, 0, 0]],
              [[2, 2, 2],
               [2, 2, 2],
               [2, 2, 2]],
              [[3, 3, 3],
               [3, 3, 3],
               [1, 1, 1]]])
n,m,o = i.shape

# compute euclidean distances of each vector to the origin
# reshape input array to 2-D, as required by cdist
# only keep diagonal, as cdist computes all pairwise distances
# reshape result, adapting it to input array and required output
d = dist.cdist(i.reshape(n*m,o),i[0]).reshape(n,m,o).diagonal(axis1=2).reshape(n,m,1)

d持有：

array([[[ 0.        ],
        [ 0.        ],
        [ 0.        ]],

       [[ 1.73205081],
        [ 1.73205081],
        [ 3.46410162]],

       [[ 3.46410162],
        [ 3.46410162],
        [ 1.73205081]]])

这种方法的最大警告是，n*m*o我们只在需要时计算距离n*m（并且它涉及大量的重塑）。

于 2013-05-10T06:44:13.420 回答
0

我正在做类似的事情，即计算视频量中每对帧的平方距离之和 (SSD)。我认为它可能对您有所帮助。

video_volume 是一个单一的 4d numpy 数组。该数组应具有维度（时间、行、列、3）和 dtype np.uint8。

输出是 dtype float 的方形 2d numpy 数组。output[i,j] 应该包含帧 i 和 j 之间的 SSD。

video_volume = video_volume.astype(float)
size_t = video_volume.shape[0]
output = np.zeros((size_t, size_t), dtype = np.float)
for i in range(size_t):
    for j in range(size_t):
        output[i, j] = np.square(video_volume[i,:,:,:] - video_volume[j,:,:,:]).sum()
于 2013-05-10T10:35:33.853 回答