实现矩阵乘法运算的代码
时间: 2023-10-26 18:22:06 浏览: 95
以下是实现矩阵乘法的 Python 代码:
```python
def matrix_multiplication(A, B):
# 获取矩阵 A 和 B 的行列数
m_A, n_A = A.shape
m_B, n_B = B.shape
# 判断两个矩阵是否可以相乘
if n_A != m_B:
print("矩阵 A 的列数不等于矩阵 B 的行数,无法相乘!")
return None
# 初始化结果矩阵 C
C = np.zeros((m_A, n_B))
# 计算矩阵乘积
for i in range(m_A):
for j in range(n_B):
for k in range(n_A):
C[i][j] += A[i][k] * B[k][j]
return C
```
其中,A 和 B 分别为两个矩阵,使用 NumPy 库进行矩阵操作。该函数首先判断两个矩阵是否可以相乘,如果可以则初始化结果矩阵 C,并进行矩阵乘法计算。最后返回结果矩阵 C。
阅读全文