NumPy transpose() 函数

NumPy 的transpose()函数会置换给定数组的维度。它会在可能的情况下返回一个视图。它常用于数据处理、科学计算和机器学习任务中，在这些任务中多维数组很常见。

默认情况下，此函数会反转数组的维度。数组根据axes的值进行转置。

如果我们想根据特定顺序转置数组，我们可以传递所需的轴顺序。例如，如果我们有一个3D数组，并且我们想交换第一个轴和最后一个轴，我们可以传递(2, 1, 0)。

语法

NumPy transpose() 函数的语法如下：

numpy.transpose(a, axes=None)

参数

以下是 NumPy transpose() 函数的参数：

• a : 要转置的数组。
• axes : 元组或整数列表，可选

返回值

transpose() 函数返回转置后的数组，其数据类型与输入数组相同。

示例 1

以下是 numpy transpose() 函数的示例，其中转置操作交换了二维数组的行和列：

import numpy as np

# Original 2D array
array_2d = np.array([[1, 2, 3], [4, 5, 6]])

# Transposing the 2D array
transposed_2d = np.transpose(array_2d)

print("Original array:\n", array_2d)
print("Transposed array:\n", transposed_2d)

输出

Original array:
 [[1 2 3]
 [4 5 6]]
Transposed array:
 [[1 4]
 [2 5]
 [3 6]]

示例 2

在这个例子中，我们将使用swapaxes()函数来交换三维ndarray中指定的轴，有效地重新排序数组的维度：

import numpy as np

# Original 3D array
array_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])

# Transposing the 3D array with specified axes
transposed_3d = np.transpose(array_3d, axes=(1, 0, 2))

print("Original array:\n", array_3d)
print("Transposed array with axes (1, 0, 2):\n", transposed_3d)

输出

Original array:
 [[[1 2]
  [3 4]]

 [[5 6]
  [7 8]]]
Transposed array with axes (1, 0, 2):
 [[[1 2]
   [5 6]]

  [[3 4]
   [7 8]]]