python中的reshape函数

Python中的reshape函数用于改变数组的形状。通过reshape函数可以将原始数组重新排列成新的形状，而不改变数组中的元素值。

使用reshape函数时，需要传入一个表示新形状的元组作为参数。该元组中的每个元素表示新数组在对应维度上的大小。

举个例子，假设有一个一维数组arr，包含18个元素。我们可以使用reshape函数将其转换为一个2行9列的二维数组，或者一个3行6列的二维数组，如下所示：

import numpy as np

arr = np.arange(18)  # 创建一个一维数组

arr_2d_1 = arr.reshape((2, 9))  # 生成一个2行9列的二维数组
arr_2d_2 = arr.reshape((3, 6))  # 生成一个3行6列的二维数组

print(arr_2d_1)
print(arr_2d_2)

输出结果：

[[ 0  1  2  3  4  5  6  7  8]
 [ 9 10 11 12 13 14 15 16 17]]
[[ 0  1  2  3  4  5]
 [ 6  7  8  9 10 11]
 [12 13 14 15 16 17]]

需要注意的是，reshape函数的参数需要满足数组元素的数量不变。如果给定的新形状与数组元素数量不匹配，将会报错。