NumPy 函数

NumPy transpose() 函数用于反转或排列数组的轴并返回修改后的数组。对于二维数组,该函数返回矩阵转置。

语法

numpy.transpose(a, axes=None) 

参数

a必填。 指定输入数组。
axes可选。 指定包含 [0,1,..,N-1] 排列的元组或列表,其中 N 是 a 的轴数。返回数组的第 ith 轴将对应于输入的编号为axes[i]的轴。如果未指定,则默认为 range(a.ndim)[::-1],它反转轴的顺序。

返回值

返回矩阵转置。

示例:矩阵转置

在下面的示例中,函数返回矩阵转置。

import numpy as np
Arr = np.array([[10,20,30],[100, 200,300]])

print("Array is:")
print(Arr)

#创建矩阵转置
print("\nMatrix transpose is:")
print(np.transpose(Arr)) 

上述代码的输出将是:

Array is:
[[ 10  20  30]
 [100 200 300]]

Matrix transpose is:
[[ 10 100]
 [ 20 200]
 [ 30 300]] 

示例:带有轴参数的 transpose()

轴参数可用于排列数组的轴。考虑以下示例。

import numpy as np
#创建一个 3-D 数组
Arr1 = np.ones((10,5,2))

#排列轴
Arr2 = np.transpose(Arr1, (2,0,1))

#检查排列数组的形状
print(Arr2.shape) 

上述代码的输出将是:

(2, 10, 5)