Python多维符号自动转置

Question

我有以下最小示例：

a = np.zeros((5,5,5))
a[1,1,:] = [1,1,1,1,1]
print(a[1,:,range(4)])

我期望输出一个 5 行 4 列的数组，其中第二行有一个数组。相反，它是一个 4 行 5 列的数组，其中第二列为 5 列。这里发生了什么，我该怎么做才能得到我期望的输出？

Answer 1

这是混合基本和高级索引的示例，如https://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#combining-advanced-and-basic-indexing中所述

切片尺寸已附加到末尾。

对于一个标量索引，这是此处描述的歧义的边缘情况。它已在之前的 SO 问题和一个或多个错误/问题中讨论过。

具有高级混合索引的 Numpy 子数组分配

在这种情况下，您可以用切片替换范围，并获得预期的顺序：

In [215]: a[1,:,range(4)].shape
Out[215]: (4, 5)               # slice dimension last
In [216]: a[1,:,:4].shape
Out[216]: (5, 4)
In [219]: a[1][:,[0,1,3]].shape
Out[219]: (5, 3)