Python设计一个矩阵相乘函数,可以完成矩阵之间的相乘计算
时间: 2024-09-24 09:30:05 浏览: 57
Python实现的矩阵转置与矩阵相乘运算示例
在Python中,我们可以使用numpy库来方便地设计一个矩阵相乘的函数,因为numpy提供了高效的矩阵运算功能。如果你没有安装numpy,可以先通过pip install numpy命令安装。下面是一个简单的例子:
```python
import numpy as np
def matrix_multiply(matrix1, matrix2):
# 验证输入是否为二维数组(即矩阵)
if not (isinstance(matrix1, np.ndarray) and isinstance(matrix2, np.ndarray)):
raise ValueError("输入必须是numpy的ndarray")
# 确保矩阵维度适配于乘法规则(行*列)
if matrix1.shape[1] != matrix2.shape[0]:
raise ValueError("矩阵维度不匹配,无法相乘")
return np.dot(matrix1, matrix2)
# 示例使用
matrix_a = np.array([[1, 2], [3, 4]])
matrix_b = np.array([[5, 6], [7, 8]])
result = matrix_multiply(matrix_a, matrix_b)
print(f"矩阵A和B的乘积结果是:\n{result}")
阅读全文