要將兩個一維數組合成一個二維數組,可以使用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()
合成二維數組,輸出的結果都是相同的。