Numpy:获取3D阵列的2D切片时遇到问题

Question

类似的问题也曾被问过，但从来没有以我需要的形式出现过

我似乎很难理解NumPy切片行为

假设我有一个512x512x120形状的Numpy数组

vol1=some_numpy_array
print(x.shape) #result: (512,512,120)

我想对这个数组取一个“z-slice”，但是我最终得到的是一个512x120数组，而不是一个512x512数组 例如，我尝试以下代码

zSlice=(vol1[:][:][2]).squeeze()
print(zSlice.shape()) #result: (512,120)

为什么生成的数组形状是(512,120)而不是(512,512)？我如何解决这个问题

Answer 1

vol1[:][:][2]的问题是： vol1[:]将给出整个数组，然后再次{}给出整个数组，最后{}给出外轴索引2到处的数组，最后{}只不过是{}，这不是您想要的

您需要获取numpy数组切片：

vol1[:, :, 2].

现在，它将获取最外层和最外层轴上的所有项目，但对于最内层轴，仅获取索引2处的项目