python - 使用另一个数组和索引数组创建 numpy 数组

Question

我有一个形状为 (m, n) 的数组 y 和一个形状为 ( m，n，k)。假设我想执行以下操作(在 Python 中):

x = np.ndarray(shape=(m,n,k))
for i in range(m):
    for j in range(n):
        for l in range(k):
            x[i,j,l] = y[indices[i,j,l],j]

有没有一种方法可以简单地使用 NumPy 来完成此操作，而不需要使用 for 循环？

Answer 1

使用 NumPy 的 advanced-indexing对于向量化分配 -

x = y[indices, np.arange(n)[:,None]]