如何更改 numpy 中矩阵对角线的值?
我检查了Numpy修改ndarray对角线,但是该函数在numpy v 1.3.0中没有实现。
假设我们有一个 np.array X,我想将对角线的所有值设置为 0。
要将 numpy 中矩阵的对角线值设置为 0,你可以使用以下方法:
import numpy as np # 创建一个示例矩阵 X = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]) # 获取矩阵 X 的对角线索引 diagonal_indices = np.diag_indices(min(X.shape)) # 将对角线元素设置为 0 X[diagonal_indices] = 0 print(X)
这将打印出矩阵 X,其中对角线上的所有元素都被设置为 0。