Numpy 笔记

numpy.transpose

假设 x=np.arange(2,2,3).reshape(2,2,3)

array([[[ 0,  1,  2],
        [ 3,  4,  5]],

       [[ 6,  7,  8],
        [ 9, 10, 11]]])

此时的数组的形状 (2,2,3) 定义为：

索引： 0 1 2

形状： 2 2 3

x = np.transpose(x, axes=(1,0,2)) 执行之后的数组如下

array([[[ 0,  1,  2],
        [ 6,  7,  8]],
 
       [[ 3,  4,  5],
        [ 9, 10, 11]]])

其实就是把元素的索引交换了，比如元素 6，它之前的索引为 (1, 0, 0)，根据 transpose 的要求，把第 0 个索引与第 1 个索引交换，因此变化后的的索引为 (0, 1, 0).

References