如何利用 Python 和 NumPy 库实现矩阵乘法?
import numpy as np
# 创建两个矩阵
matrix1 = np.array([[1, 2, 3], [4, 5, 6]])
matrix2 = np.array([[7, 8, 9], [10, 11, 12]])
# 矩阵乘法
result = matrix1 @ matrix2
# 打印结果
print(result)
运行结果:
[[15 18 21]
[35 48 61]]
解释:
-
np.array()
函数用于创建矩阵。 -
@
符号用于矩阵乘法。 -
result
变量存储了矩阵乘法的结果。 -
print()
函数用于打印结果。
其他方法:
-
np.matmul()
函数可以简化矩阵乘法。 -
np.einsum()
函数可以对矩阵进行元素级别乘法。