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 np

b = [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) * 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]])

您也可以使用diagflat, 如果b的尺寸 > 1D

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