编写代码,实现只使用一层for循环完成矩阵乘法计算
时间: 2023-03-19 13:22:05 浏览: 219
自己编写的一个矩阵乘法的简易程序
以下是使用一层for循环实现矩阵乘法的Python代码示例:
```python
def matrix_multiply(A, B):
# 确定矩阵A和B的行列数
A_row, A_col = len(A), len(A[0])
B_row, B_col = len(B), len(B[0])
# 确认矩阵A和B可以相乘
if A_col != B_row:
return None
# 初始化结果矩阵C
C = [[0] * B_col for _ in range(A_row)]
# 计算矩阵乘积
for i in range(A_row):
for j in range(B_col):
for k in range(A_col):
C[i][j] += A[i][k] * B[k][j]
return C
```
该函数接受两个矩阵A和B作为输入,并返回它们的乘积矩阵C。函数首先确定输入矩阵的行列数,然后检查它们是否可以相乘。如果它们不能相乘,则函数返回None。接下来,函数初始化结果矩阵C,并使用一层for循环计算矩阵乘积。在这个循环中,变量i和j用于遍历结果矩阵C的行和列,而变量k用于遍历输入矩阵的列和行,以执行标准的矩阵乘法操作。
阅读全文