Python中np.vstack函数的作用是什么

2024-03-28

np.vstack 函数是NumPy中的函数,用于沿着垂直方向(行方向)堆叠数组。具体来说,np.vstack 将两个或多个数组垂直堆叠在一起,形成一个新的数组。

例如,如果有两个二维数组 array1array2,它们的列数相同,可以使用 np.vstack([array1, array2]) 将它们按行堆叠在一起,生成一个新的二维数组。

import numpy as np

array1 = np.array([[1, 2, 3],
                   [4, 5, 6]])

array2 = np.array([[7, 8, 9],
                   [10, 11, 12]])

result = np.vstack([array1, array2])
print(result)

上面的代码将打印以下结果:

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

这样,np.vstack 可以帮助我们在处理数组时实现沿垂直方向的堆叠操作。