NumPy多维数组重塑：巧用transpose与reshape实现特定轴合并

本教程将深入探讨如何使用numpy的`transpose`和`reshape`函数，高效地对多维数组进行复杂重塑操作。针对需要沿特定轴合并子数组的场景，我们将通过具体示例，详细讲解如何通过轴的重新排列和维度展平，将高维数组转换为目标形状，从而实现数据的灵活组织和处理。

在数据处理和科学计算中，我们经常需要对多维数组进行灵活的重塑操作。一个常见的需求是，在一个高维数组中，我们希望将特定维度上的子数组“横向”或“纵向”地拼接起来，从而改变数组的整体形状。例如，给定一个形状为 (2, 3, 4, 5) 的4D NumPy 数组，我们可以将其理解为包含2个批次的数据，每个批次包含3个形状为 (4, 5) 的二维矩阵。我们的目标是将每个批次内的3个 (4, 5) 矩阵沿它们的最后一个维度（即列）进行水平拼接，最终得到一个形状为 (2, 4, 15) 的数组。

核心策略：transpose与reshape

虽然 NumPy 提供了 concatenate 函数用于数组拼接，但在处理这种内部子数组合并并改变整体维度结构的需求时，直接使用 concatenate 可能会比较复杂或效率不高。更优雅且强大的解决方案是结合使用 numpy.transpose 和 numpy.reshape。

• numpy.transpose: 用于重新排列数组的轴。通过改变轴的顺序，我们可以将原本不相邻的维度移动到一起，为后续的展平操作做准备。
• numpy.reshape: 用于改变数组的形状。在轴重新排列后，reshape 可以将一个或多个维度展平（合并）成一个新维度，从而实现我们所需的合并效果。

实战演练：逐步重塑数组

我们将通过一个具体的例子来演示这个过程。

准备示例数据

首先，创建一个形状为 (2, 3, 2, 2) 的4D NumPy 数组作为示例：

import numpy as np

# 定义数组维度
a1, a2, a3, a4 = 2, 3, 2, 2
# 创建一个示例数组，并重塑为 (2, 3, 2, 2)
arr = np.arange(a1 * a2 * a3 * a4).reshape((a1, a2, a3, a4))

print("原始数组 arr (形状: {}):".format(arr.shape))
print(arr)

输出的 arr 如下：

[[[[ 0  1]
   [ 2  3]]

  [[ 4  5]
   [ 6  7]]

  [[ 8  9]
   [10 11]]]


 [[[12 13]
   [14 15]]

  [[16 17]
   [18 19]]

  [[20 21]
   [22 23]]]]

我们的目标是将每个批次（第一个维度）中的3个 (2, 2) 矩阵沿它们的最后一个维度（即列）水平拼接。例如，第一个批次的第一个“行”应该变成 [0, 1, 4, 5, 8, 9]。最终数组的形状应为 (2, 2, 6)。

第一步：轴的重新排列 (transpose)

为了将 (2, 3, 2, 2) 数组中的3个 (2, 2) 子矩阵沿最后一个维度合并，我们需要将表示“3个矩阵”的维度（第二个维度，索引为1）移动到表示“2D矩阵行”的维度（第三个维度，索引为2）之后，但又在“2D矩阵列”的维度（第四个维度，索引为3）之前。

原始轴顺序是 (0, 1, 2, 3)。

• 我们希望第一个维度（批次）保持不变：0
• 我们希望第三个维度（2D矩阵的行）保持不变，但在最终输出中，它将是第二个维度：2
• 我们希望第二个维度（3个矩阵）与第四个维度（2D矩阵的列）合并，所以需要将它移到第四个维度之前：1
• 我们希望第四个维度（2D矩阵的列）保持不变，并在最终合并中与第二个维度相乘：3

因此，新的轴顺序应该是 (0, 2, 1, 3)。

腾讯智影-AI数字人

基于AI数字人能力，实现7*24小时AI数字人直播带货，低成本实现直播业务快速增增，全天智能在线直播

73

查看详情

arr_transposed = arr.transpose(0, 2, 1, 3)
print("\n转置后的数组 arr_transposed (形状: {}):".format(arr_transposed.shape))
print(arr_transposed)

转置后的数组形状变为 (2, 2, 3, 2)。观察其内容：

[[[[ 0  1]
   [ 4  5]
   [ 8  9]]

  [[ 2  3]
   [ 6  7]
   [10 11]]]


 [[[12 13]
   [16 17]
   [20 21]]

  [[14 15]
   [18 19]
   [22 23]]]]

可以看到，原本属于不同 (2, 2) 矩阵的元素，现在在 arr_transposed[0, 0] 中已经按 [0, 1], [4, 5], [8, 9] 的顺序排列，它们现在是相邻的，方便我们进行合并。

第二步：维度展平 (reshape)

现在，数组的形状是 (2, 2, 3, 2)。为了实现最终 (2, 2, 6) 的形状，我们需要将倒数第二个维度（大小为3）和最后一个维度（大小为2）合并成一个新维度，其大小为 3 * 2 = 6。

arr_reshaped = arr_transposed.reshape(a1, a3, a2 * a4) # 或 arr_transposed.reshape(2, 2, 3 * 2)
print("\n重塑后的数组 arr_reshaped (形状: {}):".format(arr_reshaped.shape))
print(arr_reshaped)

最终输出：

[[[ 0  1  4  5  8  9]
  [ 2  3  6  7 10 11]]

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

这正是我们想要的结果！第一个批次的第一个“行”是 [0, 1, 4, 5, 8, 9]。

完整操作

将上述两步结合起来，我们可以用一行代码完成：

final_arr = arr.transpose(0, 2, 1, 3).reshape(a1, a3, a2 * a4)
print("\n最终结果 (形状: {}):".format(final_arr.shape))
print(final_arr)

通用性与扩展性

这个方法对于不同维度的数组同样适用。例如，对于原始问题中提到的 (2, 3, 4, 5) 数组，我们希望将其重塑为 (2, 4, 3 * 5)。

x = np.ones((2, 3, 4, 5))
print("\n原始数组 x (形状: {}):".format(x.shape))

# 转置操作：(0, 1, 2, 3) -> (0, 2, 1, 3)
x_transposed = x.transpose(0, 2, 1, 3)
print("转置后 x_transposed (形状: {}):".format(x_transposed.shape)) # 输出: (2, 4, 3, 5)

# 重塑操作：将倒数两个维度合并 (3, 5) -> (15)
x_reshaped = x_transposed.reshape(x.shape[0], x.shape[2], x.shape[1] * x.shape[3])
print("重塑后 x_reshaped (形状: {}):".format(x_reshaped.shape)) # 输出: (2, 4, 15)

注意事项

1. 轴的顺序：确定正确的 transpose 轴顺序是解决问题的关键。这通常需要一些尝试和错误，但核心思想是：将需要合并的维度移动到相邻位置，并且位于你希望它们被展平的维度之前。
2. 视图与副本：transpose 操作通常会返回原始数组的一个视图（view），这意味着它不复制数据，而是改变数据的访问方式。然而，随后的 reshape 操作，尤其是当它涉及到对数组元素进行重新排序（即不是简单的维度重组）时，通常会创建一个新的数组副本。
3. 可读性：虽然一行代码可以完成，但在复杂的重塑场景中，分步进行 transpose 和 reshape 可以提高代码的可读性和调试的便利性。

总结

通过巧妙地结合使用 NumPy 的 transpose 和 reshape 函数，我们可以高效地解决多维数组中沿特定轴合并子数组的复杂重塑问题。这种方法提供了比 concatenate 更灵活且通常更简洁的解决方案，尤其适用于需要改变数组整体维度结构而非简单拼接的场景。掌握这种技术对于处理高维数据和进行复杂的数值计算至关重要。

以上就是NumPy多维数组重塑：巧用transpose与reshape实现特定轴合并的详细内容，更多请关注php中文网其它相关文章！