在NumPy中,reshape函數用于改變數組的形狀。
使用reshape函數的語法如下:
numpy.reshape(arr, newshape, order='C')
其中,arr是要改變形狀的數組,newshape是一個整數或整數元組,用于指定新數組的形狀。order參數可選,默認為’C’,表示按行讀取數組元素。
以下是一些使用reshape函數的示例:
import numpy as np
# 創建一個1維數組,并將其形狀改變為2x2的矩陣
arr1 = np.array([1, 2, 3, 4])
arr2 = np.reshape(arr1, (2, 2))
print(arr2)
# 輸出:
# [[1 2]
# [3 4]]
# 創建一個2x2的矩陣,并將其形狀改變為1維數組
arr3 = np.array([[1, 2], [3, 4]])
arr4 = np.reshape(arr3, (4,))
print(arr4)
# 輸出:
# [1 2 3 4]
# 創建一個3x4的矩陣,并將其形狀改變為4x3的矩陣
arr5 = np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]])
arr6 = np.reshape(arr5, (4, 3))
print(arr6)
# 輸出:
# [[ 1 2 3]
# [ 4 5 6]
# [ 7 8 9]
# [10 11 12]]
注意,reshape函數并不改變原始數組的形狀,而是返回一個改變形狀后的新數組。如果你希望改變原始數組的形狀,可以使用resize函數。