输入矩阵a和b,输出a和b的乘积
时间: 2024-02-06 09:11:13 浏览: 40
输入一个矩阵,输出一个结果
好的,以下是实现矩阵乘法的Python代码:
```python
def matrix_multiplication(a, b):
# 获取矩阵a、b的行列数
a_rows, a_cols = len(a), len(a[0])
b_rows, b_cols = len(b), len(b[0])
# 判断矩阵a、b是否可以相乘
if a_cols != b_rows:
return "Error: 矩阵a的列数必须等于矩阵b的行数"
# 初始化结果矩阵
result = [[0] * b_cols for _ in range(a_rows)]
# 矩阵乘法
for i in range(a_rows):
for j in range(b_cols):
for k in range(b_rows):
result[i][j] += a[i][k] * b[k][j]
return result
```
你可以将矩阵a和b传入这个函数,它将返回a和b的乘积矩阵。注意,矩阵a的列数必须等于矩阵b的行数才能进行乘法运算。
阅读全文