python:3d上的嵌套循环

2021-06-13 13:03:34 发布

您现在位置:Python中文网/ 问答频道 /正文

有没有办法简化三维场上的三嵌套循环?代码如下所示:

from itertools import product
for kx, ky, kz in product(freq, freq, freq):
    k = np.sqrt(kx**2+ky**2+kz**2)
    if int(k+0.5) < N/2.0:
        yaxes[field][int(k+0.5)] += A[kx][ky][kz]

A的形状是(N,N,N),freq是长度为N的特殊迭代。 也许有一个numpy工具来执行这段代码,因为这需要很长时间。你知道吗