np 矩阵拼接
时间: 2023-07-01 10:05:17 浏览: 94
对于两个矩阵 A 和 B,如果它们的维度相同,则可以通过 np.concatenate() 函数将它们沿着指定轴拼接在一起。
例如,如果我们有两个形状为 (3, 2) 的矩阵 A 和 B:
```python
import numpy as np
A = np.array([[1, 2], [3, 4], [5, 6]])
B = np.array([[7, 8], [9, 10], [11, 12]])
```
我们可以将它们沿着第二个轴拼接在一起:
```python
C = np.concatenate((A, B), axis=1)
```
这将得到形状为 (3, 4) 的新矩阵 C:
```
array([[ 1, 2, 7, 8],
[ 3, 4, 9, 10],
[ 5, 6, 11, 12]])
```
如果我们想要在不同的轴上拼接矩阵,可以指定 axis 参数。例如,如果我们要沿着第一个轴拼接矩阵:
```python
C = np.concatenate((A, B), axis=0)
```
这将得到形状为 (6, 2) 的新矩阵 C:
```
array([[ 1, 2],
[ 3, 4],
[ 5, 6],
[ 7, 8],
[ 9, 10],
[11, 12]])
```
注意,在使用 np.concatenate() 函数时,需要保证除拼接轴以外的维度相同。如果矩阵的维度不同,可以使用 np.vstack() 或 np.hstack() 函数来进行垂直或水平拼接。
阅读全文