将一维数组转换为二维数组
时间: 2023-07-26 14:40:02 浏览: 191
JavaScript实现一维数组转化为二维数组
你可以使用NumPy库中的`reshape`函数将一维数组转换为二维数组。`reshape`函数可以接受一个元组作为参数,元组中包含新数组的形状。如果你想将长度为N的一维数组转换为M行N/M列的二维数组,可以将形状设为`(M, N/M)`。例如:
```python
import numpy as np
# 创建一维数组
np_data_1d = np.array([1, 2, 3, 4, 5, 6])
# 将一维数组转换为二维数组
np_data_2d = np_data_1d.reshape((2, 3))
print(np_data_2d)
```
输出结果为:
```
[[1 2 3]
[4 5 6]]
```
在这个例子中,我们将长度为6的一维数组转换为2行3列的二维数组。使用`reshape`函数时,元组中的数字需要满足二维数组中元素的总数等于原一维数组中元素的总数。
阅读全文