如何用numpy创建一个下对角矩阵？

Question

如何使用Numpy生成动态大小的下对角矩阵？例如，如果n矩阵的大小是4我想获得这样一个矩阵：| 0 0 0 0 | | 1 0 0 0 | | 0 1 0 0 | | 0 0 1 0 |

Answer 1

我发现了使用np.eye 的最短方法：

import numpy as np

n = 4

np.eye(n, k=-1, dtype=int)

输出是：

array([[0, 0, 0, 0],

       [1, 0, 0, 0],

       [0, 1, 0, 0],

       [0, 0, 1, 0]])

Answer 2

你可以创建它，首先创建一个带零的矩阵，然后填充那些：

import numpy as np

# create matrix with zeros

n=4

mat = np.zeros((n,n))

# create indexes for where the 1s belong

rows = np.arange(1,n)

cols = np.arange(n-1)

# fill in the 1s

mat[rows, cols] = 1

输出：

[[0. 0. 0. 0.]

 [1. 0. 0. 0.]

 [0. 1. 0. 0.]

 [0. 0. 1. 0.]]