python中reshape的用法

在Python中，reshape()函数用于改变数组的形状。它可以将一个数组重新排列为指定的形状，而不改变数组中的数据。下面是两个使用reshape()函数的例子：

1. 将数组转换为一维数组：

import numpy as np

arr = np.array([[1, 2, 3], [4, 5, 6]])
new_arr = arr.reshape(-1)
print(new_arr)

输出：

[1 2 3 4 5 6]

2. 将数组转换为指定的形状：

import numpy as np

arr = np.array([1, 2, 3, 4, 5, 6])
new_arr = arr.reshape(2, 3)
print(new_arr)

输出：

[[1 2 3]
 [4 5 6]]

在第一个例子中，我们使用reshape(-1)将二维数组转换为一维数组。-1的意思是根据数组的大小自动计算维度。

在第二个例子中，我们使用reshape(2, 3)将一维数组转换为二维数组，形状为2行3列。

需要注意的是，reshape()函数返回的是一个新的数组，原始数组的形状不会改变。如果无法将数组重新排列为指定的形状，将会抛出ValueError异常。