numpy转置

numpy中的转置操作可以使用transpose()函数或者T属性来实现。

例如，对于一个二维数组a，可以使用以下代码进行转置操作：

import numpy as np

a = np.array([[1, 2], [3, 4]])
a_transpose = np.transpose(a)
a_T = a.T

print(a_transpose)
print(a_T)

输出结果为：

[[1 3]
 [2 4]]
[[1 3]
 [2 4]]

可以看到，a的转置结果为[[1 3] [2 4]]，与a.T的结果相同。

除了二维数组，numpy中的多维数组也可以进行转置操作。对于一个n维数组a，可以使用transpose()函数来指定转置的维度，例如：

import numpy as np

a = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
a_transpose = np.transpose(a, (1, 0, 2))

print(a_transpose)

输出结果为：

array([[[1, 2],
        [5, 6]],

       [[3, 4],
        [7, 8]]])

可以看到，a的转置结果为一个3维数组，第一维和第二维交换了位置，第三维保持不变。这是因为在transpose()函数中指定了维度的顺序为(1, 0, 2)。

分类