python 一维数组转二维数组,如:[1,2,3,4,5,6]变成[[1,2], [3,4], [5,6]]
时间: 2023-08-18 10:56:56 浏览: 219
Python reshape的用法及多个二维数组合并为三维数组的实例
你可以使用列表推导式和切片操作来将一维数组转换为二维数组。下面是一个示例代码:
```python
def convert_to_2d_array(arr, col):
return [arr[i:i+col] for i in range(0, len(arr), col)]
arr = [1, 2, 3, 4, 5, 6]
result = convert_to_2d_array(arr, 2)
print(result)
```
输出结果为:
```
[[1, 2], [3, 4], [5, 6]]
```
在这个示例中,`convert_to_2d_array`函数接受两个参数,`arr`表示一维数组,`col`表示每个子数组的长度。函数内部使用列表推导式和切片操作来将一维数组按照指定长度切分成多个子数组,并返回转换后的二维数组。
阅读全文