python - 减少或扩展Numpy中的广播

在下面的代码中，我们计算所有给定点对之间的向量大小。为了加快NumPy中的操作速度，我们可以使用广播

import numpy as np
points = np.random.rand(10,3)

pair_vectors = points[:,np.newaxis,:] - points[np.newaxis,:,:]
pair_dists = np.linalg.norm(pair_vectors,axis=2).shape

或外部产品迭代

it = np.nditer([points,points,None], flags=['external_loop'], op_axes=[[0,-1,1],[-1,0,1],None])
for a,b,c in it:
    c[...] = b - a
pair_vectors = it.operands[2]
pair_dists = np.linalg.norm(pair_vectors,axis=2)

我的问题是，如何使用广播或外部产品迭代来创建一个格式为10x10x6的数组，其中最后一个轴包含一对（扩展）中两个点的坐标。并且以相关的方式，可以直接使用广播或外部乘积迭代来计算对距离，即在不首先计算差向量（归约）的情况下产生形式为10x10的矩阵。

为了明确起见，以下代码使用慢循环创建所需的矩阵。

pair_coords = np.zeros(10,10,6)
pair_dists = np.zeros(10,10)
for i in range(10):
    for j in range(10):
        pair_coords[i,j,0:3] = points[i,:]
        pair_coords[i,j,3:6] = points[j,:]
        pair_dists[i,j] = np.linalg.norm(points[i,:]-points[j,:])

这是一次使用外部乘积迭代来计算距离（或应用采用一对中两个点的6个坐标并产生标量的其他任何函数）的失败尝试。

res = np.zeros((10,10))
it = np.nditer([points,points,res], flags=['reduce_ok','external_loop'], op_axes=[[0,-1,1],[-1,0,1],None])
for a,b,c in it: c[...] = np.linalg.norm(b-a)
pair_dists = it.operands[2]

最佳答案

这是一种以矢量化方式生成这些数组的方法-

from itertools import product
from scipy.spatial.distance import pdist, squareform

N = points.shape[0]

# Get indices for selecting rows off points array and stacking them
idx = np.array(list(product(range(N),repeat=2)))
p_coords = np.column_stack((points[idx[:,0]],points[idx[:,1]])).reshape(N,N,6)

# Get the distances for upper triangular elements.
# Then create a symmetric one for the final dists array.
p_dists = squareform(pdist(points))

this post中几乎没有讨论其他矢量化方法，因此也可以看看！

关于python - 减少或扩展Numpy中的广播，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/40569650/