从三维数组中提取第n个值

import numpy

arr = numpy.arange(36).reshape(3, 4, 3)

print(arr[:, :, 1])

输出

[[ 1  4  7 10]
 [13 16 19 22]
 [25 28 31 34]]

或者,

print(arr[:, :, 1].flatten())

输出

[ 1  4  7 10 13 16 19 22 25 28 31 34]

您可以找到有关numpy索引的更多信息 here .

更新

正如@madpysicalist在评论中提到的,您可以使用 ravel arr[..., 1] 访问最后一个索引,称为省略号。根据文件:

省略号扩展到:个对象的数目以进行选择 与x.ndim长度相同的元组

进一步

• Ravel vs Flatten
• Ellipsis

要从这样的数组中提取元素,可以执行以下操作:

>>> import numpy as np   
>>> arr = np.arange(36).reshape(3, 4, 3)
>>> arr[:,:,1]
array([[ 1,  4,  7, 10],
       [13, 16, 19, 22],
       [25, 28, 31, 34]])

>>> arr.flatten()
array([ 1,  4,  7, 10, 13, 16, 19, 22, 25, 28, 31, 34])