`np.stack` 是 NumPy（一个用于Python的数值计算库）中的一个函数，用于将序列中的数组沿着新的轴堆叠起来。 函数的基本形式为： ```python numpy.stack(arrays, axis=0, out=None) ``` 参数说明： * `arrays`：序列中的数组，这些数组将被堆叠。 * `axis`：堆叠的轴，0 表示垂直堆叠（在行的方向上），1 表示水平堆叠（在列的方向上）。默认为0。 * `out`：可选参数，用于指定输出数组的内存位置。 下面是一些简单的例子来说明 `np.stack` 的使用： ### 例子 1：垂直堆叠（axis=0） ```python import numpy as np a = np.array([1, 2, 3]) b = np.array([4, 5, 6]) result = np.stack((a, b)) print(result) ``` 输出： ```lua [[1 2 3] [4 5 6]] ``` ### 例子 2：水平堆叠（axis=1） ```python import numpy as np a = np.array([1, 2, 3]) b = np.array([4, 5, 6]) result = np.stack((a, b), axis=1) print(result) ``` 输出： ```lua [[1 4] [2 5] [3 6]] ``` ### 例子 3：堆叠多维数组 ```python import numpy as np a = np.array([[1, 2], [3, 4]]) b = np.array([[5, 6], [7, 8]]) result = np.stack((a, b)) print(result) ``` 输出： ```lua [[[1 2] [3 4]] [[5 6] [7 8]]] ``` 在这个例子中，我们堆叠了两个2x2的数组，结果是一个形状为(2, 2, 2)的三维数组。
