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

Question

我正在尝试创建一个 3d 数组，其单元格条目将根据单元格索引进行计算。具体来说，我想要那个 cell (i,j,k) = sqrt(i+j+k)。

使用以下 for 循环很容易做到这一点：

N=10

A=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。