NumPy transpose 的多种用法与实例 – wiki基地


深入探索 NumPy 的转置操作:transpose 的奥秘与应用

在数据科学、机器学习、图像处理以及科学计算的广阔领域中,NumPy 是一个不可或缺的强大工具。它提供了高效的多维数组对象 (ndarray) 及其配套的函数库,用于快速操作这些数组。在众多的数组操作中,转置(Transpose) 是一个基础且极其重要的概念。它指的是改变数组的维度顺序,例如将矩阵的行变成列,列变成行。

NumPy 提供了多种方式来实现转置操作,其中最核心的函数便是 numpy.transpose。本文将深入探讨 numpy.transpose 函数的多种用法、其变体(如 .T 属性、numpy.swapaxesnumpy.moveaxis)以及在实际应用中的丰富实例。

1. 理解转置(Transpose)的基本概念

转置最直观的例子是二维数组,也就是矩阵。对于一个形状为 (m, n) 的矩阵 A,其转置 AT 是一个形状为 (n, m) 的矩阵,其中 AT 的第 i 行第 j 列元素是 A 的第 j 行第 i 列元素。简单来说,就是行和列互换了。

但在 NumPy 中,数组可以拥有任意数量的维度(ndim > 2)。对于 N 维数组,转置意味着重新排列这 N 个维度的顺序。原始数组的每个元素通过其在各个维度上的索引位置来确定,转置操作就是提供一种新的方式来映射这些索引位置到新的维度顺序上。

例如,一个三维数组的形状是 (d0, d1, d2),如果我们对其进行转置,默认情况下会将其维度顺序反转,变为 (d2, d1, d0)。原数组中位于索引 [i, j, k] 的元素,在转置后的数组中会位于索引 [k, j, i]

理解转置的核心在于理解 轴(Axes) 的概念。NumPy 数组的每个维度对应一个轴,从 0 开始编号。一个形状为 (d0, d1, d2) 的三维数组有三个轴:轴 0 长度为 d0,轴 1 长度为 d1,轴 2 长度为 d2。转置就是对这些轴进行重新排列。

2. NumPy 中的转置方法概览

NumPy 提供了以下几种主要的转置或轴操作方法:

  1. .T 属性: 这是 ndarray 对象的一个属性,提供了一种快速进行默认转置(即反转轴顺序)的方式。主要用于二维数组的行/列互换。
  2. numpy.transpose(a, axes=None): 这是通用的转置函数,可以接受一个 axes 参数来指定任意的轴排列顺序。
  3. numpy.swapaxes(a, axis1, axis2): 用于交换数组中的两个指定轴的位置。
  4. numpy.moveaxis(a, source, destination): 用于将一个或多个轴从其原始位置移动到新的位置。

尽管这几种方法都能实现轴的重新排列,但它们的设计目的和用法略有不同。理解它们的区别对于选择最合适的工具至关重要。本文将重点详细讲解 numpy.transpose 及其核心参数 axes,同时也会对比其他方法。

3. ndarray.T 属性:便捷的默认转置

.T 属性是进行默认转置(反转轴顺序)的最简洁方式,尤其是在处理二维数组时。

用法: array.T

功能: 返回数组 array 的转置视图(如果可能的话)。对于 N 维数组,它相当于将轴的顺序从 (0, 1, ..., N-1) 变为 (N-1, N-2, ..., 0)

实例:

二维数组 (矩阵):

“`python
import numpy as np

创建一个二维数组 (3行 x 4列)

matrix = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])

print(“原始矩阵 shape:”, matrix.shape)
print(matrix)

使用 .T 进行转置

transposed_matrix = matrix.T

print(“\n转置后的矩阵 shape:”, transposed_matrix.shape)
print(transposed_matrix)
“`

输出:

“`
原始矩阵 shape: (3, 4)
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]

转置后的矩阵 shape: (4, 3)
[[ 1 5 9]
[ 2 6 10]
[ 3 7 11]
[ 4 8 12]]
“`

可以看到,形状从 (3, 4) 变为了 (4, 3),行和列成功互换。

一维数组:

值得注意的是,对于一维数组,.T 属性并不会改变数组的形状或内容。NumPy 的一维数组既不是行向量也不是列向量的严格表示,它只是一个序列。

“`python
vector = np.array([1, 2, 3, 4])
print(“原始向量 shape:”, vector.shape)
print(vector)

transposed_vector = vector.T
print(“\n转置后的向量 shape:”, transposed_vector.shape)
print(transposed_vector)
“`

输出:

“`
原始向量 shape: (4,)
[1 2 3 4]

转置后的向量 shape: (4,)
[1 2 3 4]
“`

这可能与线性代数中的直觉不符(将行向量变为列向量),但这是 NumPy 一维数组的特性。如果需要表示行向量或列向量,通常会使用二维数组,例如形状为 (1, N)(N, 1)

“`python

表示行向量 (1×4 形状的二维数组)

row_vector = np.array([[1, 2, 3, 4]])
print(“行向量 shape:”, row_vector.shape)
print(row_vector)

对行向量使用 .T

transposed_row_vector = row_vector.T
print(“\n转置后的行向量 shape:”, transposed_row_vector.shape)
print(transposed_row_vector)
“`

输出:

“`
行向量 shape: (1, 4)
[[1 2 3 4]]

转置后的行向量 shape: (4, 1)
[[1]
[2]
[3]
[4]]
“`

现在,我们得到了一个列向量。这表明 .T 属性对于二维或更高维度的数组是有效的。

高维数组:

对于高于二维的数组,.T 属性会将其轴顺序完全反转。

“`python

创建一个三维数组 (2x3x4)

tensor_3d = np.arange(24).reshape((2, 3, 4))
print(“原始三维数组 shape:”, tensor_3d.shape)
print(tensor_3d)

使用 .T 进行转置

transposed_tensor_3d = tensor_3d.T
print(“\n转置后的三维数组 shape:”, transposed_tensor_3d.shape)
print(transposed_tensor_3d)
“`

输出:

“`
原始三维数组 shape: (2, 3, 4)
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]

转置后的三维数组 shape: (4, 3, 2)
[[[ 0 12]
[ 4 16]
[ 8 20]]

[[ 1 13]
[ 5 17]
[ 9 21]]

[[ 2 14]
[ 6 18]
[10 22]]

[[ 3 15]
[ 7 19]
[11 23]]]
“`

原始形状 (2, 3, 4) 变为了 (4, 3, 2),轴的顺序 (0, 1, 2) 变为了 (2, 1, 0)。

.T 属性简单易用,适用于需要进行默认转置的场景。它通常返回原始数组的一个 视图(view),这意味着不复制数据,而是通过改变索引方式来访问数据,这非常高效。修改视图也会反映在原始数组上(除非视图的形状或步幅不允许直接修改)。

4. numpy.transpose(a, axes=None) 函数:灵活的轴排列

numpy.transpose 函数是进行转置的通用工具。它接受两个参数:数组 a 和可选的 axes 参数。

用法: numpy.transpose(a, axes=None)

参数:
* a: 需要进行转置操作的 NumPy 数组。
* axes: 一个由整数组成的元组或列表,其长度必须与数组 a 的维度(a.ndim)相等。这个元组指定了转置后新数组中轴的顺序。例如,如果 a 是一个三维数组,其原始轴顺序为 (0, 1, 2),那么 axes=(1, 2, 0) 表示新数组的轴 0 是原数组的轴 1,新数组的轴 1 是原数组的轴 2,新数组的轴 2 是原数组的轴 0。如果 axesNone,则执行默认转置(反转轴顺序),这与 .T 属性的行为一致。

功能: 根据 axes 参数指定的顺序重新排列数组的轴。返回转置后的数组,通常是原始数组的一个视图。

实例:

默认转置 (axes=None):

“`python
matrix = np.array([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12]])

transposed_matrix_func = np.transpose(matrix) # axes=None implicitly
print(“使用 np.transpose (默认) 的矩阵 shape:”, transposed_matrix_func.shape)
print(transposed_matrix_func)

验证与 .T 属性一致

print(“与 .T 属性结果一致:”, np.array_equal(matrix.T, transposed_matrix_func))
“`

输出:

使用 np.transpose (默认) 的矩阵 shape: (4, 3)
[[ 1 5 9]
[ 2 6 10]
[ 3 7 11]
[ 4 8 12]]
与 .T 属性结果一致: True

指定轴顺序 (axes):

这是 np.transpose 的强大之处。我们可以任意指定轴的排列。

三维数组的任意轴排列:

考虑一个形状为 (2, 3, 4) 的三维数组 tensor_3d,其轴分别为 0, 1, 2。

“`python
tensor_3d = np.arange(24).reshape((2, 3, 4))
print(“原始三维数组 shape:”, tensor_3d.shape)
print(“原始数组:\n”, tensor_3d)

示例 1: 将轴 1 移动到最前面 (新顺序: 1, 0, 2)

transposed_102 = np.transpose(tensor_3d, axes=(1, 0, 2))
print(“\n转置 (axes=(1, 0, 2)) 的 shape:”, transposed_102.shape)
print(“转置后的数组:\n”, transposed_102)

原形状 (d0, d1, d2) -> 新形状 (d1, d0, d2) -> (3, 2, 4)

示例 2: 将轴 2 移动到最前面,然后是轴 0,最后是轴 1 (新顺序: 2, 0, 1)

transposed_201 = np.transpose(tensor_3d, axes=(2, 0, 1))
print(“\n转置 (axes=(2, 0, 1)) 的 shape:”, transposed_201.shape)
print(“转置后的数组:\n”, transposed_201)

原形状 (d0, d1, d2) -> 新形状 (d2, d0, d1) -> (4, 2, 3)

“`

输出:

“`
原始三维数组 shape: (2, 3, 4)
原始数组:
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]

转置 (axes=(1, 0, 2)) 的 shape: (3, 2, 4)
转置后的数组:
[[[ 0 1 2 3]
[12 13 14 15]]

[[ 4 5 6 7]
[16 17 18 19]]

[[ 8 9 10 11]
[20 21 22 23]]]

转置 (axes=(2, 0, 1)) 的 shape: (4, 2, 3)
转置后的数组:
[[[ 0 4 8]
[12 16 20]]

[[ 1 5 9]
[13 17 21]]

[[ 2 6 10]
[14 18 22]]

[[ 3 7 11]
[15 19 23]]]
“`

如何理解 axes 参数?

axes 元组 (p0, p1, ..., p_{N-1}) 告诉 NumPy 如何构建新的数组。新数组的第 i 个轴(索引为 i)将对应于原始数组的第 p_i 个轴。换句话说,新数组的形状将是 (a.shape[p0], a.shape[p1], ..., a.shape[p_{N-1}])

例如,对于 axes=(1, 0, 2)
* 新数组的轴 0 是原数组的轴 1。
* 新数组的轴 1 是原数组的轴 0。
* 新数组的轴 2 是原数组的轴 2。

如果原数组形状是 (d0, d1, d2),那么新数组的形状就是 (d1, d0, d2)。原数组中索引为 [i, j, k] 的元素,在新数组中的索引位置如何确定呢?我们知道新数组轴 0 对应原数组轴 1,新数组轴 1 对应原数组轴 0,新数组轴 2 对应原数组轴 2。如果新数组的索引是 [x, y, z],那么这意味着它是原数组中轴 1 索引为 x,轴 0 索引为 y,轴 2 索引为 z 的元素。所以,新数组的 [x, y, z] 元素对应于原数组的 [y, x, z] 元素。这可以通过验证上面示例的输出得到印证。

np.transpose 提供了极高的灵活性来重排数组的维度,这在处理多维数据(如图像、视频、高维传感器数据等)时非常有用,因为不同库或算法可能期望特定维度的输入顺序。

5. numpy.swapaxes(a, axis1, axis2):交换两个轴

np.swapaxesnp.transpose 的一个特例,专门用于交换数组中的两个指定轴。它的代码更简洁,意图更明确,如果你只需要交换两个轴的位置,使用 np.swapaxes 会比构造 axes 元组更方便。

用法: numpy.swapaxes(a, axis1, axis2)

参数:
* a: 需要操作的数组。
* axis1: 需要交换的第一个轴的索引。
* axis2: 需要交换的第二个轴的索引。

功能: 交换数组 aaxis1axis2 轴的位置。返回交换轴后的数组视图。

实例:

交换三维数组的轴 0 和轴 2:

原始数组 tensor_3d 形状为 (2, 3, 4),轴顺序 (0, 1, 2)。交换轴 0 和轴 2 后,新轴顺序变为 (2, 1, 0)。

“`python
tensor_3d = np.arange(24).reshape((2, 3, 4))
print(“原始三维数组 shape:”, tensor_3d.shape)
print(“原始数组:\n”, tensor_3d)

交换轴 0 和轴 2

swapped_0_2 = np.swapaxes(tensor_3d, 0, 2)
print(“\n交换轴 0 和 2 后的 shape:”, swapped_0_2.shape)
print(“交换后的数组:\n”, swapped_0_2)

与 np.transpose(tensor_3d, axes=(2, 1, 0)) 比较

transposed_210 = np.transpose(tensor_3d, axes=(2, 1, 0))
print(“\n与 np.transpose(axes=(2, 1, 0)) 结果一致:”, np.array_equal(swapped_0_2, transposed_210))
“`

输出:

“`
原始三维数组 shape: (2, 3, 4)
原始数组:
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]

交换轴 0 和 2 后的 shape: (4, 3, 2)
交换后的数组:
[[[ 0 12]
[ 4 16]
[ 8 20]]

[[ 1 13]
[ 5 17]
[ 9 21]]

[[ 2 14]
[ 6 18]
[10 22]]

[[ 3 15]
[ 7 19]
[11 23]]]

与 np.transpose(axes=(2, 1, 0)) 结果一致: True
“`

交换轴 0 和轴 2 的结果与 np.transpose(axes=(2, 1, 0)) 是一致的,因为原始轴顺序是 (0, 1, 2),交换 0 和 2 后,新顺序自然是 (2, 1, 0)。

np.swapaxes 更适合于简单的轴对换场景,代码可读性更好。它也是返回视图。

6. numpy.moveaxis(a, source, destination):移动一个或多个轴

np.moveaxis 提供了另一种方式来重新排列轴,它的逻辑是将指定的轴从其原始位置移动到新的位置,同时保持其他轴的相对顺序不变。这在某些场景下比构造完整的 axes 元组更直观。

用法: numpy.moveaxis(a, source, destination)

参数:
* a: 需要操作的数组。
* source: 一个整数或由整数组成的序列,表示需要移动的轴的原始索引。
* destination: 一个整数或由整数组成的序列,表示 source 中对应轴需要移动到的目标索引。sourcedestination 的长度必须相同。目标索引可以是负数,-1 表示最后一个位置,-2 表示倒数第二个位置等。

功能:source 中的轴移动到 destination 中的位置。其他轴会相应地移动以填充空出的位置并为移动的轴腾出空间,同时保持它们原有的相对顺序。返回移动轴后的数组视图。

实例:

移动三维数组的轴:

原始数组 tensor_3d 形状为 (2, 3, 4),轴顺序 (0, 1, 2)。

“`python
tensor_3d = np.arange(24).reshape((2, 3, 4))
print(“原始三维数组 shape:”, tensor_3d.shape)
print(“原始数组:\n”, tensor_3d)

示例 1: 将轴 0 (长度为 2) 移动到位置 1 (索引 1)

原始轴顺序 (0, 1, 2)

移动轴 0 到位置 1

结果轴顺序 (1, 0, 2) – 注意:原轴 1 现在是新轴 0

moved_0_to_1 = np.moveaxis(tensor_3d, 0, 1)
print(“\n将轴 0 移动到位置 1 后的 shape:”, moved_0_to_1.shape)
print(“移动后的数组:\n”, moved_0_to_1)

原形状 (d0, d1, d2) -> 新形状 (d1, d0, d2) -> (3, 2, 4)

与 np.transpose(tensor_3d, axes=(1, 0, 2)) 比较

transposed_102 = np.transpose(tensor_3d, axes=(1, 0, 2))
print(“\n与 np.transpose(axes=(1, 0, 2)) 结果一致:”, np.array_equal(moved_0_to_1, transposed_102))

示例 2: 将轴 2 (长度为 4) 移动到位置 0 (索引 0)

原始轴顺序 (0, 1, 2)

移动轴 2 到位置 0

结果轴顺序 (2, 0, 1) – 注意:原轴 0 现在是新轴 1

moved_2_to_0 = np.moveaxis(tensor_3d, 2, 0)
print(“\n将轴 2 移动到位置 0 后的 shape:”, moved_2_to_0.shape)
print(“移动后的数组:\n”, moved_2_to_0)

原形状 (d0, d1, d2) -> 新形状 (d2, d0, d1) -> (4, 2, 3)

与 np.transpose(tensor_3d, axes=(2, 0, 1)) 比较

transposed_201 = np.transpose(tensor_3d, axes=(2, 0, 1))
print(“\n与 np.transpose(axes=(2, 0, 1)) 结果一致:”, np.array_equal(moved_2_to_0, transposed_201))
“`

输出:

“`
原始三维数组 shape: (2, 3, 4)
原始数组:
[[[ 0 1 2 3]
[ 4 5 6 7]
[ 8 9 10 11]]

[[12 13 14 15]
[16 17 18 19]
[20 21 22 23]]]

将轴 0 移动到位置 1 后的 shape: (3, 2, 4)
移动后的数组:
[[[ 0 1 2 3]
[12 13 14 15]]

[[ 4 5 6 7]
[16 17 18 19]]

[[ 8 9 10 11]
[20 21 22 23]]]

与 np.transpose(axes=(1, 0, 2)) 结果一致: True

将轴 2 移动到位置 0 后的 shape: (4, 2, 3)
移动后的数组:
[[[ 0 4 8]
[12 16 20]]

[[ 1 5 9]
[13 17 21]]

[[ 2 6 10]
[14 18 22]]

[[ 3 7 11]
[15 19 23]]]

与 np.transpose(axes=(2, 0, 1)) 结果一致: True
“`

np.moveaxis 也可以同时移动多个轴:

“`python

原始形状 (2, 3, 4, 5)

tensor_4d = np.arange(120).reshape((2, 3, 4, 5))
print(“原始四维数组 shape:”, tensor_4d.shape)

将轴 0 和轴 3 移动到末尾 (顺序保持相对不变)

原始轴 (0, 1, 2, 3)

moveaxis(tensor_4d, [0, 3], [-2, -1])

轴 0 (源) 移动到位置 -2 (目标)

轴 3 (源) 移动到位置 -1 (目标)

剩余轴 (1, 2) 保持相对顺序,填充前面

结果轴顺序 (1, 2, 0, 3)

moved_0_3_to_end = np.moveaxis(tensor_4d, [0, 3], [-2, -1])
print(“将轴 0 和 3 移动到末尾后的 shape:”, moved_0_3_to_end.shape) # (3, 4, 2, 5)

验证其等价的 transpose 调用

原始轴 (0, 1, 2, 3) -> 新轴 (1, 2, 0, 3)

transposed_1203 = np.transpose(tensor_4d, axes=(1, 2, 0, 3))
print(“与等价 transpose 结果一致:”, np.array_equal(moved_0_3_to_end, transposed_1203))
“`

输出:

原始四维数组 shape: (2, 3, 4, 5)
将轴 0 和 3 移动到末尾后的 shape: (3, 4, 2, 5)
与等价 transpose 结果一致: True

np.moveaxis 的优点在于,它通过指定“将哪个轴移动到哪个位置”来描述轴的重排,这在思维上可能比构造一个完整的 axes 序列更直接,特别是当只需要调整少数几个轴的位置时。

7. 总结转置方法的选择

  • .T 属性: 用于二维数组的简单行/列互换,或高维数组的轴顺序完全反转(等同于 np.transpose(arr, axes=None))。最简洁,适用于默认转置。
  • np.transpose(a, axes=None): 最通用和灵活的转置函数。通过 axes 参数可以实现任意的轴排列。是理解所有转置操作的基础。
  • np.swapaxes(a, axis1, axis2): 专注于交换两个指定轴的位置。如果只需要交换两个轴,它比构造 axes 更清晰。等价于 np.transpose 的一个特例。
  • np.moveaxis(a, source, destination): 通过指定轴的源位置和目标位置来重排轴。在需要将某些轴移动到特定位置,同时保持其他轴相对顺序不变时,它可能比 np.transpose 更直观。等价于 np.transpose 的另一种描述方式。

所有这些方法通常都返回原始数组的 视图,这意味着它们是内存高效的,并且修改返回的数组通常会影响原始数组(取决于内存布局)。

8. 实际应用场景中的转置

转置操作在许多计算任务中都是必不可少的步骤。以下是一些常见的应用场景:

a) 线性代数中的矩阵运算

矩阵乘法是转置最经典的用例之一。两个矩阵 A (m, n) 和 B (p, q) 可以相乘(A @ B)的条件是 n == p。如果我们需要计算 A 和 B 的转置的乘积,或者 A 的转置和 B 的乘积等等,就需要用到转置。

例如,计算矩阵与其转置的乘积 matrix @ matrix.T

“`python
matrix = np.array([[1, 2],
[3, 4],
[5, 6]]) # 3×2 矩阵

print(“原始矩阵 shape:”, matrix.shape) # (3, 2)
transposed_matrix = matrix.T
print(“转置矩阵 shape:”, transposed_matrix.shape) # (2, 3)

计算 matrix @ matrix.T (3×2 @ 2×3 -> 3×3)

product = matrix @ transposed_matrix
print(“\nmatrix @ matrix.T 的结果 shape:”, product.shape)
print(“乘积:\n”, product)

计算 matrix.T @ matrix (2×3 @ 3×2 -> 2×2)

product_T = transposed_matrix @ matrix
print(“\nmatrix.T @ matrix 的结果 shape:”, product_T.shape)
print(“乘积:\n”, product_T)
“`

输出:

“`
原始矩阵 shape: (3, 2)
转置矩阵 shape: (2, 3)

matrix @ matrix.T 的结果 shape: (3, 3)
乘积:
[[ 5 11 17]
[11 25 39]
[17 39 61]]

matrix.T @ matrix 的结果 shape: (2, 2)
乘积:
[[35 44]
[44 56]]
“`

在求解线性方程组、特征值分解、主成分分析 (PCA) 等线性代数操作中,转置是基本构建块。

b) 图像处理

图像通常表示为形状为 (高度, 宽度, 通道数) 的三维数组(例如,(H, W, C))。然而,一些图像处理库或机器学习模型可能期望通道维度在前面(例如,(C, H, W)),或者期望批量处理的图像形状为 (批量大小, 通道数, 高度, 宽度) (N, C, H, W)。这时就需要使用转置来改变轴的顺序。

假设我们有一张彩色图像数组,形状为 (480, 640, 3) (高度=480, 宽度=640, 通道=3)。

“`python

模拟一张彩色图像数据

height=480, width=640, channels=3

image_data = np.random.rand(480, 640, 3)
print(“原始图像数据 shape (HWC):”, image_data.shape)

将通道维度移动到前面 (CHW)

原始轴顺序 (0, 1, 2) -> 目标轴顺序 (2, 0, 1)

使用 np.transpose 或 np.moveaxis

image_chw_transpose = np.transpose(image_data, axes=(2, 0, 1))
print(“转置为 CHW 后的 shape:”, image_chw_transpose.shape) # (3, 480, 640)

image_chw_moveaxis = np.moveaxis(image_data, source=2, destination=0)
print(“moveaxis 为 CHW 后的 shape:”, image_chw_moveaxis.shape) # (3, 480, 640)

print(“transpose 和 moveaxis 结果一致:”, np.array_equal(image_chw_transpose, image_chw_moveaxis))

假设需要处理一个批次的图像,形状是 (批量大小, 高度, 宽度, 通道数) (N, H, W, C)

batch_size = 10
batch_images = np.random.rand(batch_size, 480, 640, 3)
print(“\n原始批次图像 shape (NHWC):”, batch_images.shape) # (10, 480, 640, 3)

某些深度学习框架(如 PyTorch)期望输入形状为 (批量大小, 通道数, 高度, 宽度) (N, C, H, W)

原始轴顺序 (0, 1, 2, 3) -> 目标轴顺序 (0, 3, 1, 2)

保持批量大小轴 (0) 不变,将通道轴 (3) 移动到位置 1,然后是原高度轴 (1) 和原宽度轴 (2)

batch_nchw_transpose = np.transpose(batch_images, axes=(0, 3, 1, 2))
print(“转置为 NCHW 后的 shape:”, batch_nchw_transpose.shape) # (10, 3, 480, 640)

使用 np.moveaxis 实现相同的操作

将轴 3 (通道) 从位置 3 移动到位置 1

batch_nchw_moveaxis = np.moveaxis(batch_images, source=3, destination=1)
print(“moveaxis 为 NCHW 后的 shape:”, batch_nchw_moveaxis.shape) # (10, 3, 480, 640)

print(“transpose 和 moveaxis 结果一致:”, np.array_equal(batch_nchw_transpose, batch_nchw_moveaxis))
“`

c) 数据预处理和特征工程

在处理表格数据时,虽然 Pandas 是更常用的工具,但有时数据可能以 NumPy 数组的形式存在,且需要按列进行操作。一个 (样本数, 特征数) 的数组,如果某些操作(如计算每行的均值或方差)需要沿特征维度(轴 1),而另一些操作(如标准化)需要沿样本维度(轴 0),则可能需要转置来方便操作。

“`python

假设数据是 (样本数, 特征数)

data = np.random.rand(100, 50) # 100 个样本,每个样本 50 个特征
print(“原始数据 shape (样本, 特征):”, data.shape) # (100, 50)

计算每个特征的均值 (沿样本轴)

feature_means = np.mean(data, axis=0)
print(“每个特征的均值 shape:”, feature_means.shape) # (50,)

如果数据是 (特征数, 样本数) 形式,计算每个特征的均值就需要沿轴 1

data_transposed = data.T # 形状 (特征, 样本) (50, 100)
print(“转置数据 shape (特征, 样本):”, data_transposed.shape) # (50, 100)

计算每个特征的均值 (沿样本轴,在新数组中是轴 1)

feature_means_transposed = np.mean(data_transposed, axis=1)
print(“转置数据计算的均值 shape:”, feature_means_transposed.shape) # (50,)

print(“两种方式计算的均值一致:”, np.allclose(feature_means, feature_means_transposed))
“`

d) 广播 (Broadcasting)

在 NumPy 中进行数组间的算术运算时,如果数组形状不兼容,NumPy 会尝试使用广播机制来使它们兼容。转置有时是调整数组形状以满足广播要求的一种方法。

例如,我们有一个形状为 (3, 4) 的矩阵和一个形状为 (4,) 的向量,想让向量与矩阵的每一列相加。直接相加会报错,或者进行的是行方向的广播。

“`python
matrix = np.arange(12).reshape((3, 4))
vector = np.arange(4)

print(“矩阵 shape:”, matrix.shape) # (3, 4)
print(“向量 shape:”, vector.shape) # (4,)

目标:让向量与矩阵的每一列相加

需要将向量视为一个列向量 (4, 1)

再将其与转置后的矩阵 (4, 3) 相加,然后转置回来

更好的方法是调整向量的形状,使其能够沿列方向广播

方法 1: 转置两次 (不太直接,但能说明转置的作用)

matrix.T shape is (4, 3)

vector shape is (4,)

(4, 3) 和 (4,) 广播规则:从末尾开始对齐,(4, 3) vs (4, 1) -> (4, 3) vs (4,) 不兼容直接相加会广播到行

我们希望 (4, 3) + (4, 1) 形式

可以将 vector 变为 (4, 1) 的列向量

vector_col = vector[:, np.newaxis] # 或 vector.reshape(-1, 1)
print(“列向量 shape:”, vector_col.shape) # (4, 1)

现在 matrix.T (4, 3) 可以与 vector_col (4, 1) 广播相加

广播规则:(4, 3) 和 (4, 1) -> 1 扩展到 3,结果形状 (4, 3)

result_transposed = matrix.T + vector_col
print(“\n(matrix.T + vector_col) shape:”, result_transposed.shape) # (4, 3)

再将结果转置回来

final_result = result_transposed.T
print(“最终结果 shape:”, final_result.shape) # (3, 4)
print(“最终结果:\n”, final_result)

方法 2: 直接使用广播(无需转置,但转置概念有助于理解如何调整形状)

matrix shape is (3, 4)

vector shape is (4,)

为了让 vector 与 matrix 的每一列相加,vector 应该能广播到形状 (3, 4),且只作用于最后一维。

这已经满足广播条件了:(3, 4) vs (4) -> (3, 4) vs (1, 4) -> (3, 4) vs (3, 4),会自动沿轴 0 广播

但是,如果我们想与每一相加,需要 (3, 4) vs (3, 1) 形式的向量

例如,如果有一个形状为 (3,) 的向量,想与矩阵的每一行相加

row_vector = np.arange(3)
print(“\n行向量 shape:”, row_vector.shape) # (3,)

直接相加会广播到列

result_row_add_col_broadcast = matrix + row_vector
print(“matrix + row_vector (广播到列) shape:”, result_row_add_col_broadcast.shape) # (3, 4)
print(“matrix + row_vector (广播到列):\n”, result_row_add_col_broadcast)

如果真的想让 (3,) 的向量与每一行相加 (即向量作为行向量广播),需要将其形状变为 (3, 1)

row_vector_col = row_vector[:, np.newaxis] # shape (3, 1)
result_row_add_row_broadcast = matrix + row_vector_col
print(“\nmatrix + row_vector[:, np.newaxis] (广播到行) shape:”, result_row_add_row_broadcast.shape) # (3, 4)
print(“matrix + row_vector[:, np.newaxis] (广播到行):\n”, result_row_add_row_broadcast)

``
在这个广播的例子中,虽然不总是直接调用
transpose来实现广播本身,但理解如何通过改变维度顺序来匹配广播规则,转置的概念是非常有帮助的。np.newaxisreshape` 是调整形状以进行广播的更常用工具,但它们的目的有时与转置相似——调整数组的维度布局。

9. 视图 (View) 与副本 (Copy) 的考虑

如前所述,arr.T, np.transpose, np.swapaxes, np.moveaxis 通常返回原始数组的 视图。视图是一个新的数组对象,但它共享原始数组的数据内存。这意味着:

  • 高效: 创建视图不需要复制大量数据,速度非常快且内存占用小。
  • 关联性: 修改视图中的元素会直接影响原始数组,反之亦然(前提是内存布局允许直接写入)。

“`python
matrix = np.array([[1, 2], [3, 4]])
transposed_view = matrix.T

print(“原始数组:\n”, matrix)
print(“转置视图:\n”, transposed_view)

修改视图中的元素

transposed_view[0, 1] = 99

print(“\n修改视图后:”)
print(“原始数组:\n”, matrix) # 原始数组也改变了
print(“转置视图:\n”, transposed_view) # 视图反映了修改
“`

输出:

“`
原始数组:
[[1 2]
[3 4]]
转置视图:
[[1 3]
[2 4]]

修改视图后:
原始数组:
[[ 1 99]
[ 3 4]]
转置视图:
[[ 1 3]
[99 4]]
“`

可以看到,修改 transposed_view[0, 1] (对应原始矩阵的 [1, 0]) 确实改变了原始矩阵的 [1, 0] 位置的元素。

什么时候可能返回副本?

在极少数情况下,如果由于轴的重排导致新的数组布局不是原始数组内存块的一个简单、连续或规则的视图,NumPy 可能需要创建一个副本来满足操作需求。但这对于标准的 transposeswapaxes 操作来说很少发生。更常见的是,对转置后的视图进行其他一些不改变元素总数但打乱内存连续性的操作(如某些切片、高级索引或不规则的 reshape)时,后续操作可能会创建副本。

如果你不希望修改转置后的结果影响原始数组,或者不确定是否返回的是视图,可以使用 .copy() 方法显式地创建一个副本:

“`python
matrix = np.array([[1, 2], [3, 4]])
transposed_copy = matrix.T.copy() # 显式创建副本

print(“原始数组:\n”, matrix)
print(“转置副本:\n”, transposed_copy)

修改副本中的元素

transposed_copy[0, 1] = 99

print(“\n修改副本后:”)
print(“原始数组:\n”, matrix) # 原始数组没有改变
print(“转置副本:\n”, transposed_copy)
“`

输出:

“`
原始数组:
[[1 2]
[3 4]]
转置副本:
[[1 3]
[2 4]]

修改副本后:
原始数组:
[[1 2]
[3 4]]
转置副本:
[[ 1 3]
[99 4]]
“`

在这种情况下,修改副本不会影响原始数组,因为它们占用了不同的内存空间。

10. 常见误区和注意事项

  • 一维数组的转置: 如前所述,一维数组的 .Tnp.transpose 不会改变其形状。如果需要行向量或列向量,请使用形状为 (1, N)(N, 1) 的二维数组。
  • axes 参数的理解: axes=(p0, p1, ..., p_{N-1}) 表示新数组的第 i 个轴对应于原始数组的第 p_i 个轴。多花时间练习和可视化不同 axes 值对形状和元素位置的影响非常重要。
  • 视图与副本: 大多数时候是视图,高效。但要注意修改视图可能会影响原始数组。如果需要独立的操作结果,请使用 .copy().
  • 与其他形状操作的结合: 转置经常与其他形状操作(如 reshape, squeeze, expand_dims)结合使用,以获得最终所需的数组布局。例如,将 (H, W) 图像变为 (1, H, W, 1) 用于 CNN 输入。

11. 结语

NumPy 的转置操作是处理多维数组的基石。.T 属性提供了便捷的默认转置,而 np.transpose 凭借其灵活的 axes 参数能够实现任意的轴排列。np.swapaxesnp.moveaxis 则提供了在特定场景下更直观的轴操作方式。熟练掌握这些工具,并理解它们如何影响数组的形状和元素的索引方式,对于高效地进行数据处理、科学计算以及构建复杂的算法模型至关重要。

希望本文通过详细的解释、丰富的实例和对不同函数的对比,帮助您深入理解 NumPy 转置操作的各个方面,并在未来的学习和工作中游刃有余地运用它们。


发表评论

您的邮箱地址不会被公开。 必填项已用 * 标注

滚动至顶部