自定义矩阵行与列实现矩阵相加
时间: 2024-06-17 17:07:49 浏览: 83
实现矩阵相加,需要先确保两个矩阵的行数和列数相等。然后,对应位置上的元素相加即可。
下面是自定义矩阵行与列实现矩阵相加的示例代码:
```python
class Matrix:
def __init__(self, m, n):
self.m = m
self.n = n
self.matrix = [[0 for j in range(n)] for i in range(m)]
def __add__(self, other):
if self.m != other.m or self.n != other.n:
raise ValueError("Matrix size doesn't match")
res = Matrix(self.m, self.n)
for i in range(self.m):
for j in range(self.n):
res.matrix[i][j] = self.matrix[i][j] + other.matrix[i][j]
return res
```
在上面的代码中,我们定义了一个 Matrix 类,用于表示一个矩阵。在初始化时,需要传入矩阵的行数 m 和列数 n,并初始化一个 m 行 n 列的矩阵。
接下来,我们重载了 __add__ 方法,用于实现矩阵相加操作。在相加之前,我们需要先检查两个矩阵的行数和列数是否相等,如果不相等,则抛出异常。然后,我们创建一个新的矩阵 res,用于存储相加的结果。最后,我们使用两层循环遍历每个位置上的元素,将其相加,并存储到 res 中。
阅读全文