将 numpy 3-d 矩阵切片为 2-d 矩阵答案

【问题标题】：Slicing a numpy 3-d matrix into 2-d matrix将 numpy 3-d 矩阵切片为 2-d 矩阵
【发布时间】：2018-07-18 14:23:24
【问题描述】：

我有一个 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]]])

我想提取一个二维矩阵，以便将结果沿 3 维矩阵的列切片。比如：

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

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

感谢您的帮助。

【问题讨论】：

标签： python numpy matrix pytorch

【解决方案1】：

这只是最后一个暗淡，带有转置：

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

【讨论】：

【解决方案2】：

您可以组合切片、整形和转置，如下所示：

t[:, :, 1:].reshape((2, 2)).T

希望对你有帮助

【讨论】：