如何通过for循环将多个NumPy数组堆叠为三维数组？

要实现将多个二维NumPy数组堆叠成目标三维数组，核心是确保每次迭代时数组维度匹配，沿着**第0轴（最外层维度）**进行堆叠。以下是两种基于for循环的实现思路：

方法一：初始化空数组后循环填充 #

先创建一个形状匹配的空三维数组，再逐个将二维数组扩展维度后堆叠进去：

import numpy as np

a1 = np.array([[1,1,1],[1,1,1],[1,1,1]]) 
a2 = np.array([[2,2,2],[2,2,2],[2,2,2]])
a3 = np.array([[3,3,3],[3,3,3],[3,3,3]])
a4 = np.array([[4,4,4],[4,4,4],[4,4,4]])
a5 = np.array([[5,5,5],[5,5,5],[5,5,5]])

# 将所有待堆叠数组存入列表，方便循环遍历
arr_list = [a1, a2, a3, a4, a5]

# 初始化空三维数组：形状为(0, 3, 3)，与单个二维数组的维度对齐
result = np.empty((0, 3, 3))

for arr in arr_list:
    # 给当前二维数组新增一个维度（从(3,3)变为(1,3,3)），再沿第0轴堆叠
    result = np.vstack([result, arr[np.newaxis, :, :]])

print(result)

方法二：从第一个数组开始循环追加 #

以第一个数组为初始三维数组，后续逐个追加剩余数组：

import numpy as np

a1 = np.array([[1,1,1],[1,1,1],[1,1,1]]) 
a2 = np.array([[2,2,2],[2,2,2],[2,2,2]])
a3 = np.array([[3,3,3],[3,3,3],[3,3,3]])
a4 = np.array([[4,4,4],[4,4,4],[4,4,4]])
a5 = np.array([[5,5,5],[5,5,5],[5,5,5]])

# 剩余待堆叠的数组列表
arr_list = [a2, a3, a4, a5]
# 将第一个二维数组扩展为三维数组，作为初始结果
result = a1[np.newaxis, :, :]

for arr in arr_list:
    # 扩展维度后沿第0轴堆叠
    result = np.vstack([result, arr[np.newaxis, :, :]])

print(result)

补充：无需循环的简化写法 #

如果不限制必须用for循环，NumPy内置函数可以一行实现需求：

# 使用np.stack直接在第0轴堆叠
result = np.stack([a1, a2, a3, a4, a5], axis=0)

# 或者用列表推导式配合np.vstack
result = np.vstack([arr[np.newaxis, :, :] for arr in [a1,a2,a3,a4,a5]])

关键细节说明 #

• arr[np.newaxis, :, :]的作用是给二维数组新增一个维度，让原本(3,3)的形状变为(1,3,3)，确保能和三维数组在第0轴上完成堆叠。
• np.vstack专门用于沿垂直方向（第0轴）堆叠数组，最终生成的三维数组中，每个原二维数组对应最外层维度的一个切片。

内容的提问来源于stack exchange，提问作者Eric Hwang