使用numpy可以获得三维空间中所有对象相对于一个对象的位置

Question

我想得到numpy数组中向量对的所有排列之间的差异。在

在我的特定用例中，这些向量是对象列表的3D位置向量。在

因此，如果我有一个数组r = [r1, r2, r3]，其中r1，r2和{}是三维向量，我想要以下内容：

[[r1-r1 r1-r2 r1-r3]
 [r2-r1 r2-r2 r2-r3]
 [r3-r1 r3-r2 r3-r3]]

其中-op按元素顺序应用于向量。在

基本上，向量等价于：

^{pr2}$

然而，outer函数似乎在减法之前将我的向量数组展平，然后再对其进行整形。例如：

>>> vectors = np.arange(6).reshape(2, 3) # Two 3-dimensional vectors
>>> print(vectors)
[[0 1 2]
 [3 4 5]]

>>> results = np.subtract.outer(vectors, vectors)
>>> print(results.shape)
(2, 3, 2, 3)

我期待的结果是：

>>> print(result)
[[[ 0  0  0]
  [-3 -3 -3]]
 [[ 3  3  3]
  [ 0  0  0]]]

>>> print(result.shape)
(2, 2, 3)

不迭代上面的数组我能实现吗？在

Answer 1

答案几乎总是broadcasting：

>>> r = np.arange(6).reshape(2, 3)
>>> r[:, None] - r
array([[[ 0,  0,  0],
        [-3, -3, -3]],

       [[ 3,  3,  3],
        [ 0,  0,  0]]])

索引中的None与np.newaxis相同，并将大小为1的维度添加到数组的形状中。所以你要从一个形状为(2, 1, 3)的数组中减去一个形状为(2, 3)的数组，它通过广播转换成(1, 2, 3)，最终结果就是你想要的(2, 2, 3)数组。虽然广播在概念上类似于使用np.tile或np.repeat，但它更有效，因为它避免创建原始数组的扩展副本。在