python - 将 numpy 3 维矩阵切片为 2 维矩阵

Question

我有一个 3d numpy 矩阵 t 如下，随机生成:

t = np.random.rand(2,2,2)
array([[[ 0.80351862,  0.25631294],
        [ 0.7971346 ,  0.29468456]],

       [[ 0.33771957,  0.91776256],
        [ 0.6018604 ,  0.55290615]]])

我想提取一个 2-d 矩阵，以便将结果沿着 3-d 矩阵的列进行切片。像这样的东西:

array([[ 0.25631294  ,  0.91776256],
       [ 0.29468456,  0.55290615]])

我怎样才能以这种方式切片？

感谢您的帮助。

Answer 1

这只是使用最后一个暗淡的转置:

>>> t[:,:,1].T
array([[ 0.25631294,  0.91776256],
       [ 0.29468456,  0.55290615]])