温馨提示×

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

小亿
143
2023-10-11 00:26:44
栏目: 编程语言

要将两个一维数组合成一个二维数组,可以使用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()合成二维数组,输出的结果都是相同的。

0