输入两个 𝑛 n 行 𝑚 m 列的矩阵 𝐴 A 和 𝐵 B,输出它们的和 𝐴 + 𝐵 A+B,矩阵加法的规则是两个矩阵中对应位置的值进行加和,具体参照样例。python
时间: 2024-09-20 09:17:57 浏览: 62
数据结构-三元组稀疏矩阵求和函数
在Python中,如果你有一个二维列表表示矩阵,你可以通过遍历矩阵的每个元素并简单地将它们相加来计算两个矩阵的和。这里是一个简单的函数示例:
```python
def matrix_addition(A, B):
# 确保矩阵A和B有相同的行数和列数
assert len(A) == len(B), "Matrices must have the same number of rows."
assert all(len(row) == len(B[0]) for row in A), "Matrices must have the same number of columns."
# 初始化一个新的矩阵,用于存储结果
result = [[0] * len(B[0]) for _ in range(len(A))]
# 对应元素相加
for i in range(len(A)):
for j in range(len(A[0])):
result[i][j] = A[i][j] + B[i][j]
return result
# 示例:
# A = [[1, 2], [3, 4]]
# B = [[5, 6], [7, 8]]
# C = matrix_addition(A, B)
# 输出将是:[[6, 8], [10, 12]]
```
这个函数首先检查输入矩阵是否具有相同的维度,然后逐个计算每个元素的和,并将其存储在新的`result`矩阵中。
阅读全文