在NumPy中,可以使用reshape()函數來改變數組的形狀。以下是使用reshape()函數改變數組形狀的示例:
import numpy as np
# 創建一個一維數組
arr = np.array([1, 2, 3, 4, 5, 6])
# 將一維數組轉換為二維數組
arr_2d = arr.reshape(2, 3)
print(arr_2d)
# 輸出:
# [[1 2 3]
# [4 5 6]]
# 創建一個二維數組
arr_2d = np.array([[1, 2, 3], [4, 5, 6]])
# 將二維數組轉換為一維數組
arr_1d = arr_2d.reshape(6)
print(arr_1d)
# 輸出:
# [1 2 3 4 5 6]
# 可以使用-1來自動推斷數組的形狀
arr_1d = arr_2d.reshape(-1)
print(arr_1d)
# 輸出:
# [1 2 3 4 5 6]
# 創建一個三維數組
arr_3d = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
# 將三維數組轉換為二維數組
arr_2d = arr_3d.reshape(4, 2)
print(arr_2d)
# 輸出:
# [[1 2]
# [3 4]
# [5 6]
# [7 8]]
在使用reshape()函數改變數組形狀時,需要確保新形狀的總元素個數與原數組的總元素個數相等,否則會拋出ValueError異常。