python输入矩阵a和b,输出a和b的乘积。
时间: 2024-02-13 20:00:37 浏览: 77
好的,以下是Python实现矩阵乘法的代码:
```python
def matrix_multiply(a, b):
"""
计算矩阵a和矩阵b的乘积
"""
# 获取矩阵a的行数和列数
a_row, a_col = len(a), len(a[0])
# 获取矩阵b的行数和列数
b_row, b_col = len(b), len(b[0])
# 判断矩阵a和矩阵b是否可以相乘
if a_col != b_row:
return "矩阵a的列数必须等于矩阵b的行数"
# 初始化结果矩阵
result = [[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):
result[i][j] += a[i][k] * b[k][j]
return result
# 输入矩阵a和b
a = [[1, 2], [3, 4]]
b = [[5, 6], [7, 8]]
# 计算矩阵a和矩阵b的乘积
c = matrix_multiply(a, b)
# 输出结果
for row in c:
print(row)
```
代码中定义了一个`matrix_multiply()`函数,用于计算两个矩阵的乘积。函数中首先判断两个矩阵是否可以相乘,如果不能相乘则返回错误信息。然后初始化结果矩阵,通过三重循环计算矩阵乘积。最后,我们输入矩阵a和b,调用`matrix_multiply()`函数计算它们的乘积,并输出结果。
阅读全文