我有以下最小示例:
a = np.zeros((5,5,5))
a[1,1,:] = [1,1,1,1,1]
print(a[1,:,range(4)])
我希望输出一个 5 行 4 列的数组,其中我们在第二行有一个。相反,它是一个具有 4 行和 5 列的数组,其中一个在第二列。这里发生了什么,我该怎么做才能获得预期的输出?
最佳答案
这是混合基本索引和高级索引的示例,如 https://docs.scipy.org/doc/numpy/reference/arrays.indexing.html#combining-advanced-and-basic-indexing 中所述。
切片维度已附加到末尾。
对于一个标量索引,这是此处描述的歧义的边缘情况。它已在之前的 SO 问题和一个或多个错误/问题中进行了讨论。
Numpy sub-array assignment with advanced, mixed indexing
在这种情况下,您可以用切片替换范围,并获得预期的顺序:
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)
关于Python多维符号自动转置,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/45471197/