python - 直观地解释这个 4D numpy 数组索引

Question

x = np.random.randn(4, 3, 3, 2)
print(x[1,1])

output:
[[ 1.68158825 -0.03701415]
[ 1.0907524  -1.94530359]
[ 0.25659178  0.00475093]]

我是 python 新手。我真的无法理解上面的 4-D 数组索引。 x[1,1] 是什么意思？

例如，对于向量

a = [[2][3][8][9]], a[0] = 2, a[3] = 9. 

我明白了，但我不知道 x[1,1] 指的是什么。

请详细说明。谢谢。

Answer 1

二维数组是一个矩阵:数组的数组。

4D 数组基本上是矩阵的矩阵:

指定一个索引给你一个矩阵数组:

>>> x[1]
array([[[-0.37387191, -0.19582887],
        [-2.88810217, -0.8249608 ],
        [-0.46763329,  1.18628611]],

       [[-1.52766397, -0.2922034 ],
        [ 0.27643125, -0.87816021],
        [-0.49936658,  0.84011388]],

       [[ 0.41885001,  0.16037164],
        [ 1.21510322,  0.01923682],
        [ 0.96039904, -0.22761806]]])

指定两个索引给你一个矩阵:

>>> x[1, 1]
array([[-1.52766397, -0.2922034 ],
       [ 0.27643125, -0.87816021],
       [-0.49936658,  0.84011388]])

指定三个索引给你一个数组:

>>> x[1, 1, 1]
array([ 0.27643125, -0.87816021])

指定四个索引给你一个元素:

>>> x[1, 1, 1, 1]
-0.87816021212791107

x[1,1] 为您提供保存在大矩阵第 2 行第 2 列中的小矩阵。