切片NumPy三维数组

Question

我正在学习NumPy。我创建了这样的三维阵列：

a = np.array([[[1,2], [3,4]], [[5,6], [7,8]]])
a 
a
# array([[[1, 2],
#         [3, 4]],
#        [[5, 6],
#         [7, 8]]])

我正在尝试访问下面的值3到6：

a[0,1] # array([3, 4])
a[1,0] # array([5, 6])
a[0,1][1,0] # IndexError: too many indices for array

这可能通过一个线性切片实现吗？

Answer 1

试试这个：

import numpy as np
a = np.array([[[1,2,3], [4,5,6], [7,8,9]]])
tmp = a.flatten()
tmp[2:6]