reshape函数怎么使用
时间: 2023-06-17 21:02:36 浏览: 157
Reshaping data with the reshape package
在Python中,reshape()函数是用于改变矩阵或数组形状的函数。它通常用于机器学习中的数据预处理和数据变换过程中。reshape()函数需要一个参数作为新的形状,该参数可以是一个整数元组或一个列表,以指定新的形状。例如:
```
import numpy as np
# 创建一个 2x4 的数组
arr = np.array([[1, 2, 3, 4], [5, 6, 7, 8]])
# 改变数组形状为 4x2
new_arr = arr.reshape((4, 2))
print(new_arr)
```
输出:
```
array([[1, 2],
[3, 4],
[5, 6],
[7, 8]])
```
在上面的代码中,我们使用了numpy库中的reshape()函数,将原始数组的形状从2x4改变为4x2。注意,在改变形状时,新的形状的大小必须与原始数组的大小相同,否则会报错。
阅读全文