python将一维数组变道固定的大小的二维数组里
时间: 2024-03-08 14:48:42 浏览: 55
python将三维数组展开成二维数组的实现
5星 · 资源好评率100%
可以使用numpy库中的reshape函数将一维数组转换为二维数组。例如,有一个一维数组a,长度为n,我们希望将其转换为一个m行k列的二维数组b,那么可以使用如下代码:
```python
import numpy as np
a = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9])
n = len(a)
m, k = 3, 3
b = a.reshape((m, k))
```
其中,reshape函数的参数为一个元组,元组中的每个元素表示对应维度的长度。在上面的例子中,reshape((m, k))表示将一维数组a转换为一个3行3列的二维数组b。注意,转换前后数组的元素个数必须相等,即n=m*k。如果不相等,将会抛出异常。
阅读全文