可以使用numpy庫中的reshape函數將一維數組轉換為二維數組。假設一維數組名為arr,需要轉換為m行n列的二維數組,則可以使用如下代碼:
import numpy as np
arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
m = 3 # 行數
n = 3 # 列數
arr_2d = np.reshape(arr, (m, n))
print(arr_2d)
輸出結果為:
[[1 2 3]
[4 5 6]
[7 8 9]]
也可以使用列表推導式來實現一維數組轉二維數組的功能,如下所示:
arr = [1, 2, 3, 4, 5, 6, 7, 8, 9]
m = 3 # 行數
n = 3 # 列數
arr_2d = [arr[i:i+n] for i in range(0, len(arr), n)]
print(arr_2d)
輸出結果為:
[[1, 2, 3], [4, 5, 6], [7, 8, 9]]