如何访问/引用 Numpy 数组的特定维度

Question

假设我有一个 3-D 数组 x，我知道我可以做到：

x[:,ind,:] = y

但是如果 x 是一个 N 维数组并且我想在它的第 M 维中放置一个索引 (ind)。

我尝试像x.take(ind, axis=M) 一样使用numpy.take，但这会返回切片数组的副本而不是引用。

Numpy 是否为此提供任何功能？

Answer 1

您可以先使用slice 创建索引，然后再应用它：

>>> index = [slice(None)]*N
>>> index[M] = ind
>>> x[index] = y