矢量顺序numpy计算

@Psidom方法的变体:

np.tensordot(d, r, axes=((0,1,2), (1,2,3)))

这取决于 tensordot documentation :

给定两个张量(维数大于或等于1的数组), a 和 b (a_axes, b_axes) 一 的和 b a_axes 和 b_axes .

从根本上讲 张索多 在两个阵列中 。与@Psidom的方法相比,它没有提供更好的性能。

另外,请参见 Understanding tensordot

你可以这样矢量化它;自从 d 一维是否小于 r d 将播出 axis=0 属于 从而避免循环;从那以后 r 是布尔数组, d[r].sum() 将与相同 (d*r).sum :

(d * r).sum(axis=(1,2,3))
# array([ 775.17049697,  728.61537246,  735.05686655,  765.19469927,
#         759.44834287])

x :

((d*r).sum(axis=(1,2,3)) == x).all()
# True