小能豆

更改 numpy 中矩阵对角线的值

python

如何更改 numpy 中矩阵对角线的值?

我检查了Numpy修改ndarray对角线,但是该函数在numpy v 1.3.0中没有实现。

假设我们有一个 np.array X,我想将对角线的所有值设置为 0。


阅读 78

收藏
2024-05-13

共1个答案

小能豆

要将 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。

2024-05-13