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

Question

假设我有一个三维数组x，我知道我可以这样做：

x[:,ind,:] = y

但是如果x是N-D数组并且我想在其第M维中放入索引（ind），该怎么办。

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

Numpy会为此提供任何功能吗？

Answer 1

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

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