Numpy的transpose()代码理解

numpy.transpose 函数是对 NumPy 数组进行转置操作的函数。转置通常指的是将矩阵的行变成列，或者将列变成行。
对于二维数组（即矩阵），使用 numpy.transpose 或其等效的 T 属性可以得到数组的转置。以下是一个例子：

import numpy as np
# 创建一个二维数组
arr = np.array([[1, 2, 3],
                [4, 5, 6]])
# 使用 transpose 函数进行转置
transposed_arr = np.transpose(arr)
print(transposed_arr)
# 输出:
# [[1 4]
#  [2 5]
#  [3 6]]
# 使用 T 属性进行转置
transposed_arr = arr.T
print(transposed_arr)
# 输出:
# [[1 4]
#  [2 5]
#  [3 6]]

对于更高维度的数组，numpy.transpose 函数允许你指定转置的轴。例如，对于一个三维数组，你可以指定如何交换轴：

# 创建一个三维数组
arr_3d = np.array([[[1, 2], [3, 4]],
                   [[5, 6], [7, 8]]])
# 指定轴进行转置，例如交换第一个和第二个轴
transposed_arr_3d = np.transpose(arr_3d, (1, 0, 2))
print(transposed_arr_3d)
# 输出:
# [[[1 2]
#   [5 6]]
#
#  [[3 4]
#   [7 8]]]

在上面的例子中，np.transpose(arr_3d, (1, 0, 2)) 表示将三维数组的第一个轴和第二个轴交换位置，第三个轴保持不变。