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