如何使用numpy创建下对角线矩阵

Question

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

| 0 0 0 0 |
| 1 0 0 0 |
| 0 1 0 0 |
| 0 0 1 0 |

Answer 1

您可以创建它，方法是首先创建一个包含0的矩阵，然后填充1：

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

Answer 2

我找到了使用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]])