如何利用 Python 和 NumPy 库实现矩阵乘法?

如何利用 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() 函数可以对矩阵进行元素级别乘法。
相似内容
更多>