用numpy执行外部加法

Question

抱歉，如果这是一个愚蠢的问题，但是我刚开始使用python /
numpy，我真的不确定最有效的处理方法。我正在为一些学生准备一个演示N体模拟器，但是现在，我正在通过循环这些粒子的位置来计算粒子之间的力，这可以想象与糖蜜一样慢。基本上，给定一个向量x[i]，我想计算：

n[i] = sum from j = 0 to n-1, j != i of (x[i]-x[j])^-2,

使用numpy函数而不是循环。如果可以执行外部加法/乘法：

m[i,j] = x[i]-x[j],

m[i,j] = x[i]*x[j],

我可以用它来进行计算。

Answer 1

所有带有两个输入参数的通用函数都有一个属性outer：

x = np.array([1, 2, 3])
np.subtract.outer(x, x)

给出：

array([[ 0, -1, -2],
       [ 1,  0, -1],
       [ 2,  1,  0]])

和

np.multiply.outer(x, x)

结果是：

array([[1, 2, 3],
       [2, 4, 6],
       [3, 6, 9]])