在单位矩阵中用零替换一 |麻木 |

【问题标题】：replace one with zero in identity matrix | Numpy |在单位矩阵中用零替换一 |麻木 |
【发布时间】：2021-11-13 14:15:47
【问题描述】：

我正在学习 Numpy。我如何从单位矩阵中获得预期的输出

单位矩阵

1, 0, 0
0, 1, 0
0, 0, 1

预期输出：

1, 0, 0
0, 0, 0
0, 0, 0

【问题讨论】：

【解决方案1】：

您的问题尚不清楚，但假设您有一个零矩阵，对角线上为 1，并且希望除第一个元素外全为零：

a = np.array([[1,0,0],
              [0,1,0],
              [0,0,1]])
np.fill_diagonal(a, 0)
a[0,0] = 1

输出：

array([[1., 0., 0.],
       [0., 0., 0.],
       [0., 0., 0.]])

【讨论】：