numpy填充矩阵对角线值与另一个矩阵

Question

这是用另一个完整矩阵填充矩阵对角元素的更简单方法吗？

b = [1,2,3,4,5,6,7,8,9]

a = np.zeros((9, 9), int)

np.fill_diagonal(a, b)

我希望结果是

[[1. 0  0  ...........0 0]

 [0. 2. 0  ...........0 0]

 [0. 0. 3. ...  0  0  0 0]

 [0. 0. 0. 4 0  0  0  0 0]

 [0. 0. 0. 0 5  0  0  0 0]

 [0. 0. 0. ...  6  0  0 0]

 [0. 0. 0. ...  0  7  0 0]

 [0. 0. 0. ...  0  0  8 0]

 [0. 0. 0. ...  0  0  0 9]]

Answer 1

你的方法确实有效：import numpy as npb = [1,2,3,4,5,6,7,8,9]a = np.zeros((9, 9), int)np.fill_diagonal(a, b)替代：a[np.diag_indices_from(a)] = b

Answer 2

这是其中一件事numpy.diag：a = numpy.diag(b)

Answer 3

只是为了好玩，np.eye有广播。np.eye(a.shape[0], dtype=int) * barray([[1, 0, 0, 0, 0, 0, 0, 0, 0],       [0, 2, 0, 0, 0, 0, 0, 0, 0],       [0, 0, 3, 0, 0, 0, 0, 0, 0],       [0, 0, 0, 4, 0, 0, 0, 0, 0],       [0, 0, 0, 0, 5, 0, 0, 0, 0],       [0, 0, 0, 0, 0, 6, 0, 0, 0],       [0, 0, 0, 0, 0, 0, 7, 0, 0],       [0, 0, 0, 0, 0, 0, 0, 8, 0],       [0, 0, 0, 0, 0, 0, 0, 0, 9]])您也可以使用diagflat, 如果b的尺寸 > 1Dnp.diagflat(b)# np.diagflat([b])# np.diagflat(np.array([b]))array([[1, 0, 0, 0, 0, 0, 0, 0, 0],       [0, 2, 0, 0, 0, 0, 0, 0, 0],       [0, 0, 3, 0, 0, 0, 0, 0, 0],       [0, 0, 0, 4, 0, 0, 0, 0, 0],       [0, 0, 0, 0, 5, 0, 0, 0, 0],       [0, 0, 0, 0, 0, 6, 0, 0, 0],       [0, 0, 0, 0, 0, 0, 7, 0, 0],       [0, 0, 0, 0, 0, 0, 0, 8, 0],       [0, 0, 0, 0, 0, 0, 0, 0, 9]])