将单位矩阵|Numpy|中的一替换为零

Question

我正在学习Numpy。如何从身份矩阵获得预期输出

身份矩阵

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

预期输出：

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

Answer 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.]])