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