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

44次阅读

共计 517 个字符，预计需要花费 2 分钟才能阅读完成。

要将两个一维数组合成一个二维数组，可以使用 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)

输出结果为：