温馨提示×

numpy中怎么改变数组形状

小亿
85
2024-02-02 09:44:34
栏目: 编程语言

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

0