当前位置:首页 > 编程技术 > 正文

函数如何返回一个二维数组

函数如何返回一个二维数组

```pythondef create_2d_array(rows, cols, value : return [[value] cols for _ in range...

```python

def create_2d_array(rows, cols, value):

return [[value] cols for _ in range(rows)]

使用示例

result = create_2d_array(3, 4, 0) 创建一个3x4的二维数组,所有元素都是0

print(result)

```

这个函数`create_2d_array`接受三个参数:`rows`表示行数,`cols`表示列数,`value`表示二维数组中每个元素的值。函数内部使用嵌套的列表推导式来创建一个二维数组,其中外层列表推导式创建行,内层列表推导式创建列。

```python

def create_2d_array_with_values(rows, cols):

return [[i cols + j for j in range(cols)] for i in range(rows)]

使用示例

result = create_2d_array_with_values(3, 4)

print(result)

```

在这个例子中,二维数组中的每个元素都是其行索引和列索引的乘积。

最新文章