Python NumPy reshape方法用法及代碼示例

例子

從 1D 轉換為 2D

考慮以下一維數組：

a = np.array([1,2,3,4,5,6])
a



array([1, 2, 3, 4, 5, 6])

要將其更改為 2 行 3 列的 2D 數組：

np.reshape(a, (2,3))



array([[1, 2, 3],
       [4, 5, 6]])

要將一維數組轉換為隻有一個元素的二維數組：

np.reshape(a, (1, a.size))



array([[1, 2, 3, 4, 5, 6]])

從 2D 轉換為 1D

假設我們有以下二維數組：

a = np.array([[1,2],[3,4]])
a



array([[1, 2],
       [3, 4]])

要獲得一維表示：

np.reshape(a, 4)



array([1, 2, 3, 4])

在這裏，我們可以簡單地使用 ravel(~) 方法來展平數組：

a = np.array([[1,2],[3,4]])
a.ravel()



array([1, 2, 3, 4])

推斷形狀

shape 中的值 -1 告訴 NumPy 推斷合適的形狀。例如：

np.reshape([1,2,3,4], [-1,1])



array([[1],
       [2],
       [3],
       [4]])

我們可以手動指定 [4,-1] 作為形狀，但結果行數可以從以下方式推斷：

• 數組中值的數量

• 列數(在本例中為1)。