numpy-ndarray - 切片 NumPy 3d 数组

Question

我正在学习 NumPy。我创建了这样的 3d 数组：

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

这可以通过单线切片实现吗？

标签： numpy-ndarraynumpy-slicing

Answer 1

尝试这个：

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

tmp = a.flatten()
tmp[2:6]