如何从多个2D numpy数组按列交错堆叠得到指定维度的新数组

问题原因说明 #

你使用np.column_stack没有达到预期的原因是：np.column_stack([arr1, arr2, arr3])会直接按数组先后顺序拼接所有列，得到的列顺序是「arr1全部列 → arr2全部列 → arr3全部列」，和你需要的交错列顺序不符。

实现方案 #

假设你的三个20×308数组分别命名为arr1、arr2、arr3，可以用以下两种方法实现需求：

方法1：stack + reshape（最简洁）

import numpy as np

# 核心操作
result = np.stack([arr1, arr2, arr3], axis=2).reshape(20, -1)

# 验证输出形状
print(result.shape) # 输出 (20, 924)

原理：

• 先通过np.stack在第三维度堆叠三个数组，得到形状为(20, 308, 3)的中间数组，每个位置(行, 列, 数组序号)对应原数组的对应元素
• 调用reshape(20, -1)时，numpy默认按行优先顺序展平最后两维，同一列位置的三个数组元素会按顺序拼接，正好符合你要的列交错规则。

方法2：切片赋值（可读性更高，支持自定义顺序）

如果需要灵活调整列的排列规则，可以直接用切片指定位置赋值：

result = np.empty((20, 924), dtype=arr1.dtype)
result[:, 0::3] = arr1 # 输出数组第0、3、6...列赋值为arr1的对应列
result[:, 1::3] = arr2 # 输出数组第1、4、7...列赋值为arr2的对应列
result[:, 2::3] = arr3 # 输出数组第2、5、8...列赋值为arr3的对应列

内容的提问来源于stack exchange，提问作者Tayyab Khalil