矩阵乘法python源码
时间: 2024-12-24 22:13:48 浏览: 4
矩阵乘法在Python中通常通过numpy库来进行,因为其提供了高效的数学运算支持。以下是简单的二维数组(可以视为矩阵)相乘的Python代码示例:
```python
import numpy as np
# 定义两个矩阵
matrix_a = np.array([[1, 2], [3, 4]])
matrix_b = np.array([[5, 6], [7, 8]])
# 使用numpy的dot函数进行矩阵乘法
result_matrix = np.dot(matrix_a, matrix_b)
print(result_matrix)
```
如果你想要手动实现矩阵乘法(适用于较小规模的矩阵),可以按照线性代数的规则进行逐元素计算,例如:
```python
def manual_matrix_multiply(matrix1, matrix2):
if len(matrix1[0]) != len(matrix2): # 检查是否可以相乘
raise ValueError("Matrices cannot be multiplied")
result = [[0] * len(matrix2[0]) for _ in range(len(matrix1))]
for i in range(len(matrix1)):
for j in range(len(matrix2[0])):
for k in range(len(matrix2)):
result[i][j] += matrix1[i][k] * matrix2[k][j]
return result
# 如果你想手动计算
manual_result = manual_matrix_multiply(matrix_a.tolist(), matrix_b.tolist())
print(manual_result)
```
阅读全文