np.vstack
时间: 2023-07-04 16:09:54 浏览: 118
python numpy–数组的组合和分割实例
`np.vstack` 是 NumPy 库中的一个函数,用于垂直(沿着行的方向)堆叠数组。具体来说,它将传入的数组沿着第一个轴(即行)连接起来,返回一个新的数组。
例如,如果我们有两个形状相同的二维数组 `a` 和 `b`:
```
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
```
使用 `np.vstack` 将它们垂直堆叠起来:
```
c = np.vstack((a, b))
```
得到的数组 `c` 的形状为 `(4, 2)`,内容为:
```
array([[1, 2],
[3, 4],
[5, 6],
[7, 8]])
```
注意,传入 `np.vstack` 的参数应该是一个元组,即使只有一个数组也要写成 `(a,)` 的形式。
阅读全文