python两个一维数组怎么合成二维数组

640
2023/10/4 0:26:44
栏目: 编程语言
开发者测试专用服务器限时活动,0元免费领,库存有限,领完即止! 点击查看>>

要将两个一维数组合成一个二维数组,可以使用numpy库中的函数np.stack()np.vstack()

使用np.stack()时,需要指定axis参数的值为0,表示按垂直方向(行)堆叠数组。

以下是示例代码:

import numpy as np
# 两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
# 使用np.stack()合成二维数组
arr_2d = np.stack((arr1, arr2), axis=0)  # axis=0表示按行堆叠
print(arr_2d)

输出结果为:

[[1 2 3]
[4 5 6]]

使用np.vstack()时,直接传入需要合并的一维数组即可。

以下是示例代码:

import numpy as np
# 两个一维数组
arr1 = np.array([1, 2, 3])
arr2 = np.array([4, 5, 6])
# 使用np.vstack()合成二维数组
arr_2d = np.vstack((arr1, arr2))
print(arr_2d)

输出结果为:

[[1 2 3]
[4 5 6]]

无论使用np.stack()还是np.vstack()合成二维数组,输出的结果都是相同的。

辰迅云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>

推荐阅读: Python列表矩阵转置怎么实现