np.array如何合并?
时间: 2024-12-06 19:15:50 浏览: 16
在NumPy中,可以使用`np.concatenate()`, `np.vstack()`, 或 `np.hstack()` 函数来合并数组。这里简要解释一下这三个函数:
1. **np.concatenate(arrs, axis=0)**: 这个函数用于沿着给定的轴(axis,默认为0,即行方向)堆叠多个数组。例如:
```python
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
result = np.concatenate((arr1, arr2))
```
2. **np.vstack(tup)** 或 **np.stack(arrs, axis=0)**: 这两个函数都是垂直堆叠数组,相当于沿列方向合并,适合一维变多维的情况。比如:
```python
arr1 = np.array([[1], [2]])
arr2 = np.array([[3], [4]])
result = np.vstack((arr1, arr2))
```
3. **np.hstack(tup)** 或 **np.column_stack(arrs)**: 这两个函数则是水平堆叠数组,沿行方向合并一维数组,变成二维数组。如:
```python
arr1 = np.array([1, 2])
arr2 = np.array([3, 4])
result = np.hstack((arr1, arr2))
```
使用以上任一函数时,确保输入的数组在维度兼容的情况下才能顺利堆叠。
阅读全文