np.hstack,np.vstack
时间: 2024-05-01 11:14:15 浏览: 8
np.hstack和np.vstack是NumPy库中的两个函数,用于在水平和垂直方向上对数组进行堆叠操作。
np.hstack函数用于将多个数组按水平方向进行堆叠,即将它们连接成一个更大的数组。这意味着数组的列数将增加。要求堆叠的数组在水平方向上具有相同的行数。
np.vstack函数用于将多个数组按垂直方向进行堆叠,即将它们连接成一个更大的数组。这意味着数组的行数将增加。要求堆叠的数组在垂直方向上具有相同的列数。
这两个函数都可以接受一个元组或列表作为参数,其中包含要堆叠的数组。以下是使用np.hstack和np.vstack的示例:
```python
import numpy as np
# 创建两个示例数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5, 6], [7, 8]])
# 使用np.hstack进行水平堆叠
hstack_result = np.hstack((a, b))
print("Horizontal Stack Result:")
print(hstack_result)
# 使用np.vstack进行垂直堆叠
vstack_result = np.vstack((a, b))
print("Vertical Stack Result:")
print(vstack_result)
```
输出结果:
```
Horizontal Stack Result:
[[1 2 5 6]
[3 4 7 8]]
Vertical Stack Result:
[[1 2]
[3 4]
[5 6]
[7 8]]
```