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)。