在NumPy中,可以使用reshape()函數來改變數組的形狀。reshape()函數接受一個元組作為參數,該元組表示新數組的形狀。
以下是一個例子:
import numpy as np
# 創建一個1維數組
arr = np.array([1, 2, 3, 4, 5, 6])
# 將1維數組轉換為2行3列的二維數組
new_arr = arr.reshape((2, 3))
print(new_arr)
輸出結果為:
[[1 2 3]
[4 5 6]]
reshape()函數還可以使用-1作為參數,表示自動計算該維度的大小。例如,如果想將一個數組轉換為一個有3行的二維數組,可以這樣寫:
import numpy as np
# 創建一個1維數組
arr = np.array([1, 2, 3, 4, 5, 6])
# 將1維數組轉換為有3行的二維數組
new_arr = arr.reshape((3, -1))
print(new_arr)
輸出結果為:
[[1 2]
[3 4]
[5 6]]