np.concatenate用法和参数
时间: 2024-05-16 07:18:44 浏览: 200
np.concatenate是NumPy库中的函数,用于将两个或多个数组沿指定轴连接起来。
它的语法如下:
```python
np.concatenate((array1, array2, ...), axis=0)
```
参数说明:
- `array1, array2, ...`:要连接的数组序列。
- `axis`:指定连接的轴。默认为0,表示沿着第一个维度连接,也可以是一个整数或元组,来指定其他维度。
例如,我们有两个一维数组arr1和arr2:
```python
import numpy as np
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
```
我们可以使用`np.concatenate`将它们连接在一起:
```python
result = np.concatenate((arr1, arr2))
print(result)
```
输出:
```
[1 2 3 4 5 6]
```
我们还可以通过指定`axis`参数来连接多维数组。例如,对于两个二维数组arr3和arr4:
```python
arr3 = np.array([[1, 2], [3, 4]])
arr4 = np.array([[5, 6]])
```
我们可以在轴0上连接它们:
```python
result = np.concatenate((arr3, arr4), axis=0)
print(result)
```
输出:
```
[[1 2]
[3 4]
[5 6]]
```
我们也可以在轴1上连接它们:
```python
result = np.concatenate((arr3, arr4.T), axis=1)
print(result)
```
输出:
```
[[1 2 5]
[3 4 6]]
```
这样,我们就可以使用np.concatenate来连接不同维度的数组。
阅读全文