numpy中怎么改变数组形状

在NumPy中，可以使用reshape()函数来改变数组的形状。reshape()函数接受一个元组作为参数，该元组表示新数组的形状。

以下是一个例子：

import numpy as np  # 创建一个1维数组 arr = np.array([1, 2, 3, 4, 5, 6])  # 将1维数组转换为2行3列的二维数组 new_arr = arr.reshape((2, 3))  print(new_arr) 

输出结果为：

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

reshape()函数还可以使用-1作为参数，表示自动计算该维度的大小。例如，如果想将一个数组转换为一个有3行的二维数组，可以这样写：

import numpy as np  # 创建一个1维数组 arr = np.array([1, 2, 3, 4, 5, 6])  # 将1维数组转换为有3行的二维数组 new_arr = arr.reshape((3, -1))  print(new_arr) 

输出结果为：

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