python - 缺少元素的矢量化求和

Question

我想解决以下和s ，这是按元素定义的:



其中 z 的元素和 lambda属于复杂类型和 N是实数和向量的长度。这可以在循环中完成，但效率低下。这是我的解决方案:

import numpy as np

N = 10
z = np.random.rand(N)
lamb = np.random.rand(N)

s = []
indexs = np.arange(N)

for idx in indexs:

    summand = np.sum(1/(z[idx] - lamb[indexs[indexs!=idx]]))

    s.append(summand/N)

s = np.array(s)

谁能指点我一个完整的方向 numpy解决方案？

Answer 1

import numpy as np

N = 10
z = np.random.rand(N)
lamb = np.random.rand(N)
ss = 1 / np.subtract.outer(z, lamb)
np.fill_diagonal(ss, 0)
ss = ss.sum(axis=1) / N