应用函数实现矩阵乘法
时间: 2023-07-16 22:15:34 浏览: 42
矩阵乘法可以用函数实现,下面是一个简单的示例代码:
```python
def matrix_multiply(A, B):
# 确定矩阵 A 和 B 的行和列
m, n = len(A), len(A[0])
p, q = len(B), len(B[0])
# 验证矩阵是否可以相乘
if n != p:
print("Error: 矩阵 A 的列数不等于矩阵 B 的行数")
return None
# 初始化结果矩阵 C
C = [[0 for j in range(q)] for i in range(m)]
# 矩阵乘法计算
for i in range(m):
for j in range(q):
for k in range(n):
C[i][j] += A[i][k] * B[k][j]
return C
```
这个函数接受两个矩阵作为参数,返回它们的积。在函数内部,我们首先确定输入矩阵 A 和 B 的行和列,然后验证它们是否可以相乘。如果不能相乘,函数将返回 None。然后,我们初始化结果矩阵 C,并使用三个循环计算矩阵乘积。最后,我们返回结果矩阵 C。
下面是一个使用这个函数的示例:
```python
A = [[1, 2], [3, 4]]
B = [[5, 6], [7, 8]]
C = matrix_multiply(A, B)
print(C)
```
这将输出结果矩阵:
```
[[19, 22], [43, 50]]
```
阅读全文