numpy 中reshape函数用法
时间: 2023-10-22 16:05:24 浏览: 95
numpy库reshape用法详解
numpy 中的 reshape 函数用于改变数组的形状。具体用法如下:
numpy.reshape(a, newshape, order='C')
其中:
- a:要改变形状的数组;
- newshape:新的形状,可以是一个整数或一个整数元组;
- order:可选参数,表示重新排列数组的顺序。默认值为 'C',即按行排列。
示例:
```python
import numpy as np
# 创建一个 3x4 的二维数组
a = np.arange(12).reshape(3, 4)
print(a)
# 输出:
# [[ 0 1 2 3]
# [ 4 5 6 7]
# [ 8 9 10 11]]
# 将二维数组转换为一维数组
b = a.reshape(12)
print(b)
# 输出:
# [ 0 1 2 3 4 5 6 7 8 9 10 11]
# 将一维数组转换为 2x6 的二维数组
c = b.reshape(2, 6)
print(c)
# 输出:
# [[ 0 1 2 3 4 5]
# [ 6 7 8 9 10 11]]
# 将二维数组转换为 4x3 的二维数组,并按列排列
d = a.reshape(4, 3, order='F')
print(d)
# 输出:
# [[ 0 3 6]
# [ 4 7 10]
# [ 8 11 2]
# [ 1 5 9]]
```
阅读全文