numpy怎么实现在某个维度上拼接?
时间: 2024-02-25 22:57:10 浏览: 56
利用numpy实现一、二维数组的拼接简单代码示例
在numpy中,可以使用`numpy.concatenate()`函数在某个维度上拼接数组。该函数接受一个元组作为参数,用于指定沿哪个维度进行拼接。下面是一个示例代码:
```python
import numpy as np
# 创建两个数组
a = np.array([[1, 2], [3, 4]])
b = np.array([[5], [6]])
# 在第二个维度上拼接数组
c = np.concatenate((a, b), axis=1)
print(c)
```
输出结果为:
```
array([[1, 2, 5],
[3, 4, 6]])
```
在上面的代码中,我们创建了两个数组`a`和`b`,然后使用`numpy.concatenate()`函数在第二个维度上拼接了这两个数组,得到了一个新的数组`c`。注意,我们将两个数组放入了一个元组中作为函数的第一个参数,然后使用`axis=1`指定了拼接的维度。
阅读全文