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]]