np.concatenate
时间: 2024-01-07 12:04:02 浏览: 52
np.concatenate 是 NumPy 库中用于连接(拼接)数组的函数。它可以按照给定的轴或维度将多个数组进行拼接,生成一个新的数组。
函数语法如下:
```python
numpy.concatenate((a1, a2, ...), axis=0, out=None)
```
其中,参数 `a1, a2, ...` 是要连接的数组序列,`axis` 指定拼接的轴或维度(默认为0),`out` 用于指定输出结果的数组。
例如,对于如下两个一维数组:
```python
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5, 6])
```
可以使用 `np.concatenate` 将它们拼接成一个新的一维数组:
```python
c = np.concatenate((a, b))
print(c) # [1 2 3 4 5 6]
```
也可以指定拼接的轴为 0,生成一个二维数组:
```python
d = np.concatenate((a.reshape(1, -1), b.reshape(1, -1)), axis=0)
print(d)
"""
[[1 2 3]
[4 5 6]]
"""
```
需要注意的是,要拼接的数组的形状必须相同,否则会引发错误。
阅读全文