Python代码生成二维数组
时间: 2024-08-08 19:01:02 浏览: 46
在Python中,生成二维数组(也称为二维列表或矩阵)有很多种方式。以下是几种常见的方法:
1. **直接赋值**:
```python
n = 3 # 数组大小
mat = [ * n for _ in range(n)]
```
这里用列表推导式创建了一个全零的n x n矩阵。
2. **使用numpy库**:
```python
import numpy as np
mat = np.zeros((n, n))
```
Numpy提供了快速高效的数据处理能力,`np.zeros`函数创建一个全零数组。
3. **初始化填充特定值**:
```python
mat = [['a'] * n for _ in range(n)]
for i in range(n):
mat[i][i] = 'b' # 在主对角线放特定值
```
可以通过循环自定义矩阵的内容。
4. **从文件读取或用户输入**:
```python
rows = int(input("请输入行数:"))
mat = []
for _ in range(rows):
row = input().split()
mat.append([int(i) for i in row])
```
从用户输入获取每一行元素并组成矩阵。
5. **列表嵌套列表的方式** (更适用于已知固定结构的数据):
```python
data = [(val1, val2, ...) for _ in range(n) for _ in range(n)]
mat = [data[i:i+n] for i in range(0, len(data), n)]
```
根据实际需求选择合适的方法。
阅读全文