矩阵的对角线可以通过NumPy中的np.diag函数获得。如果想要在计算矩阵乘法时忽略对角线,可以使用np.fill_diagonal函数将对角线元素设置为0。以下是示例代码:
import numpy as np
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
diagonal = np.diag(matrix)
np.fill_diagonal(matrix, 0)
result = np.dot(matrix, np.array([1, 2, 3]))
print(result)
上一篇:不要进入指令。
下一篇:不要计算零,但保留组号。