将一维数组转化为二维数组
时间: 2023-06-24 13:07:35 浏览: 99
JavaScript实现一维数组转化为二维数组
将一维数组转化为二维数组需要确定二维数组的行数和列数,然后按照一定的规律将一维数组中的元素赋值给二维数组。以下是一个示例代码:
```python
def convert_to_2d_array(arr, rows, cols):
if rows * cols != len(arr):
return None
else:
return [arr[i:i+cols] for i in range(0, len(arr), cols)]
# 示例
arr = [1, 2, 3, 4, 5, 6]
rows = 2
cols = 3
result = convert_to_2d_array(arr, rows, cols)
print(result) # [[1, 2, 3], [4, 5, 6]]
```
在上面的代码中,我们首先判断了一维数组和二维数组的元素个数是否匹配,如果不匹配则返回 None。接下来使用列表推导式将一维数组中的元素赋值给二维数组。列表推导式中使用了切片的方式将一维数组划分为长度为 cols 的子数组,然后按顺序将这些子数组添加到二维数组中。最后返回转化后的二维数组。
阅读全文