Numpy transpose() 函数
Numpy transpose() 函数用于对给定数组的维度进行置换。它会尽可能地返回一个视图。它常用于数据处理、科学计算和机器学习等多维数组较为常见的任务中。
默认情况下,此函数会反转数组的维度。数组会根据轴的值进行转置。
如果我们想要按特定顺序转置数组,可以传入所需的轴的顺序。例如,如果我们有一个三维数组,并且想要交换第一个轴和最后一个轴,我们可以传递 (2, 1, 0)。
语法
Numpy transpose() 函数的语法如下:-
numpy.transpose(a,axes=None)
参数
以下是 Numpy transpose() 函数的参数:-
- a : 需要转置的数组。
- axes : 元组或整数列表,可选
返回值
transpose() 函数返回转置后的数组,其数据类型与输入数组相同。
示例 1
以下是 numpy transpose() 函数的示例,其中转置操作会交换二维数组的行和列 -
import numpy as np # 原始二维数组 array_2d = np.array([[1, 2, 3], [4, 5, 6]]) # 转置二维数组 transposed_2d = np.transpose(array_2d) print("原始数组: ", array_2d) print("转置数组: ", transposed_2d)
输出
原始数组: [[1 2 3] [4 5 6]] 转置数组: [[1 4] [2 5] [3 6]]
示例 2
在此示例中,我们将使用 swapaxes() 函数交换三维 ndarray 中指定的轴,从而有效地重新排序数组的维度 -
import numpy as np # 原始三维数组 array_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]]) # 使用指定轴转置三维数组 transposed_3d = np.transpose(array_3d, axis=(1, 0, 2)) print("原始数组: ", array_3d) print("轴为 (1, 0, 2) 的转置数组: ", transposed_3d)
输出
原始数组: [[[1 2] [3 4]] [[5 6] [7 8]]] 轴为 (1, 0, 2) 的转置数组: [[[1 2] [5 6]] [[3 4] [7 8]]]