使用另一个数组和索引数组创建 numpy 数组答案

【问题标题】：Create numpy array with another array and an indexing array使用另一个数组和索引数组创建 numpy 数组
【发布时间】：2017-11-09 23:42:23
【问题描述】：

我有一个形状为(m, n) 的数组y 和一个形状为(m, n, k) 的数组indices。假设我想做以下事情（在 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 循环？

【问题讨论】：

【解决方案1】：

使用 NumPy 的 advanced-indexing 进行矢量化赋值 -

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

【讨论】：