python堆叠不同长度数组
时间: 2023-10-15 09:08:25 浏览: 82
python将三维数组展开成二维数组的实现
要堆叠不同长度的数组,可以使用numpy库中的vstack函数。vstack函数可以将一系列不同长度的数组垂直堆叠在一起,返回一个新的数组。
示例如下:
```python
import numpy as np
a = np.array([1, 2, 3])
b = np.array([4, 5])
c = np.array([6])
result = np.vstack((a, b, c))
print(result)
```
输出结果为:
```
[[1 2 3]
[4 5]
[6]]
```
在这个例子中,我们定义了三个不同长度的一维数组a、b、c。我们将它们传递给vstack函数,它会将它们垂直堆叠在一起,并返回一个新的二维数组result。
注意,使用vstack函数时,传递的数组必须具有相同的维度。在本例中,我们传递的是一维数组,但它们被视为二维数组,并且在垂直方向上堆叠在一起。
阅读全文