numpy 矩阵拼接
时间: 2024-12-03 18:14:41 浏览: 13
在NumPy库中,矩阵拼接通常通过`numpy.concatenate()`函数或`numpy.vstack()`和`numpy.hstack()`函数来完成。这两个函数可以水平拼接(`hstack`)一维数组或垂直拼接(`vstack`)二维数组。
- `numpy.concatenate(arrs, axis=0)`:沿着指定轴(axis)将输入的一系列数组连接起来。如果`axis=0`(默认),则是按行拼接;如果`axis=1`,则按列拼接。
- `numpy.vstack(tupple_of_arrays)`:垂直堆叠数组,相当于`concatenate([array_1, array_2, ...], axis=0)`。
- `numpy.hstack(tupple_of_arrays)`:水平堆叠数组,相当于`concatenate([array_1, array_2, ...], axis=1)`。
例如,如果你有两个一维数组需要拼接成一个高维度数组,或者有多个二维数组想沿某一方向拼接,你可以这样做:
```python
import numpy as np
# 横向拼接两个数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
merged_h = np.hstack((arr1, arr2))
# 纵向拼接两个数组
arr3 = np.array([[7, 8], [9, 10]])
arr4 = np.array([[11, 12], [13, 14]])
merged_v = np.vstack((arr3, arr4))
print("Horizontal Merge:", merged_h)
print("Vertical Merge:", merged_v)
阅读全文