Python如何实现矩阵输入
时间: 2024-09-20 15:04:40 浏览: 41
使用python制作黑客帝国矩阵雨
在Python中,有几种常见的方式来实现矩阵(二维数组)的输入。这里我会介绍两种常见的方法:
1. **列表嵌套**:
Python本身支持直接使用列表来表示矩阵。你可以让用户逐行输入元素,然后用这些元素创建一个二维列表。
```python
def input_matrix(rows):
matrix = []
for _ in range(rows):
row = list(map(int, input("请输入一行元素,用空格分隔: ").split()))
matrix.append(row)
return matrix
rows = int(input("请输入矩阵的行数: "))
matrix = input_matrix(rows)
```
在这个例子中,`input()`函数用于获取用户的输入,`split()`分割字符串成单个数字,再用`map()`函数转换为整数。
2. **numpy库**:
如果你需要更高效的操作和数值计算,可以使用NumPy库,它提供了专门的数据结构`numpy.ndarray`来表示矩阵。
```python
import numpy as np
rows = int(input("请输入矩阵的行数: "))
columns = int(input("请输入矩阵的列数: "))
# 使用numpy从用户输入的逗号分隔值创建矩阵
matrix_string = input("请输入矩阵,每行元素用逗号分隔: ")
matrix = np.array(matrix_string.split("\n"), dtype=int).reshape(rows, columns)
print(matrix)
```
在这个例子中,`reshape()`函数用来根据用户输入的行数和列数调整数据的形状。
阅读全文