Python中np.vstack函数怎么使用

2024-03-28

np.vstack函数用于垂直堆叠数组。它将输入数组沿垂直方向堆叠在一起,返回一个新的数组。

使用方法如下:

import numpy as np

# 创建两个数组
arr1 = np.array([[1, 2, 3],
                 [4, 5, 6]])

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

# 使用np.vstack函数垂直堆叠这两个数组
result = np.vstack((arr1, arr2))

print(result)

运行结果为:

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