根据单元格索引填充 numpy 数组

Question

我正在尝试创建一个 3d 数组，其单元格条目将根据单元格索引进行计算。具体来说，我想要那个 cell (i,j,k) = sqrt(i+j+k)。使用以下 for 循环很容易做到这一点：N=10A=np.zeros((N,N,N))for i in range(N):    for j in range(N):        for k in range(N):            A[i][j][k] = np.sqrt(i+j+k)我想知道 numpy 是否具有使这些嵌套的 for 循环变得多余的内置函数。

Answer 1

最简单和高性能的方法是使用开放网格np.ogrid，然后执行相关操作 -

I,J,K = np.ogrid[:N,:N,:N]
A = np.sqrt(I+J+K)

或者np.sum对于单线开放网格的广播总和 -

A = np.sqrt(np.sum(np.ogrid[:N,:N,:N]))

相关：General workflow on vectorizing loops involving range iterators。