numpy的reshape函数示例
时间: 2023-10-27 19:11:15 浏览: 85
numpy库reshape用法详解
numpy的reshape函数用于将数组重塑为新的形状。下面是一个示例:
```python
import numpy as np
# 创建一个一维数组
a = np.array([1, 2, 3, 4, 5, 6])
# 将一维数组重塑为二维数组
b = a.reshape(2, 3)
print(a)
print(b)
```
输出:
```
[1 2 3 4 5 6]
[[1 2 3]
[4 5 6]]
```
这个示例中,我们首先创建了一个一维数组a,包含6个元素。然后,我们使用reshape函数将一维数组a重塑为一个2x3的二维数组b。最后,我们打印出a和b的值,可以看到a是一个一维数组,而b是一个2x3的二维数组,元素值与a相同。
阅读全文