Python如何获取numpy三维数组的元素?

1 个回答|81 次阅读
Coco老师 - 科普专家
擅长:编程教育

从三维数组中访问元素,我们可以使用逗号分隔的整数来表示元素的维度和索引。

访问第一个二维数组中的第二个数组的第三个元素:

import numpy as np
arr = np.array([[[1, 2, 3], [4, 5, 6]], [[7, 8, 9], [10, 11, 12]]])
print(arr[0, 1, 2])

执行结果:

6

过程分析:

arr[0,1,2] 中的第一个数字表示第一个维度,包含两个数组,分别是:

[[1, 2, 3], [4, 5, 6]]
和:
[[7, 8, 9], [10, 11, 12]]

这里第一个数字值是0,也就是索引是0的数组,所以,应该选择[[1, 2, 3], [4, 5, 6]]这个数组;

然后,arr[0,1,2] 中第二个数字表示第二个维度,它又包含连个数组,分别是:

[1,2,3]
和:
[4,5,6]

这个第二个数字值是1,也就是索引是1的数组,所以,应该选择[4,5,6]这个数组;

最后是arr[0,1,2] 中的第三个数字表示第三个维度,其中包含三个数值:

4
5
6

这个第三个数字值是2,表示索引是2的数值,所以,最终结果是 6 。