python – 迭代一个带有两个隐式循环的数组

2023年2月5日 248次阅读

是否可以在具有两个索引的数组上隐式迭代？

这是我想做的一个非常简单的例子：

import numpy as np

x = np.arange(3)
y = np.zeros(3)

for i in range(3):
    y[i] = np.sum(x - x[i])

有一个隐式循环(总和)和一个显式循环(对于范围(3)中的i)…是否有可能有一个完全隐式的版本？

最佳答案 如果可能的话,你应该总是尝试在计算机科学之前使用数学.您的表达式y [i] = np.sum(x – x [i])可以用一个小代数重写为y [i] = np.sum(x) – x.size * x [i].这使得您可以在没有任何循环的情况下重写代码非常清楚：

y = np.sum(x) - x.size * x

显而易见,对于大型阵列,它运行速度比@JoshAdel的解决方案快得多,对于1000大小的输入,x400更快：

>>> x = np.random.normal(size=(1000,))
>>> np.allclose(np.sum(x - x[:,None], 1), np.sum(x) - x.size * x)
True

%timeit np.sum(x - x[:,None], 1)
100 loops, best of 3: 6.33 ms per loop

%timeit np.sum(x) - x.size * x
100000 loops, best of 3: 16.5 us per loop
点赞