在使用numpy时如何改变数组的形状
时间: 2023-11-27 21:50:33 浏览: 74
numpy 矩阵形状调整:拉伸、变成一位数组的实例
在使用NumPy时,可以使用`reshape()`函数来改变数组的形状。`reshape()`函数接受一个元组作为参数,该元组指定了新形状的维度。
例如,假设有一个一维数组,我们想将其转换为一个二维数组,其中有3行和4列,可以像这样使用`reshape()`函数:
```
import numpy as np
# 创建一个长度为12的一维数组
a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12])
# 将一维数组转换为二维数组
b = a.reshape((3, 4))
print(b)
```
输出结果为:
```
[[ 1 2 3 4]
[ 5 6 7 8]
[ 9 10 11 12]]
```
其中,`reshape()`函数的参数是一个元组`(3, 4)`,指定了新数组的维度为3行4列。
阅读全文