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

发布时间：2020-12-20 11:30:17 所属栏目：Python 来源：网络整理

导读：是否可以在具有两个索引的数组上隐式迭代？这是我想做的一个非常简单的例子： import numpy as npx = np.arange(3)y = np.zeros(3)for i in range(3): y[i] = np.sum(x - x[i]) 有一个隐式循环(总和)和一个显式循环(对于范围(3)中的i)…是否有可能有一个完

是否可以在具有两个索引的数组上隐式迭代？
这是我想做的一个非常简单的例子：

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[:,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

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!