NumPy的diag方法使用案例
编辑日期: 2024-11-28 文章阅读: 次
NumPy的diag方法使用案例
创建一个下对角线为1、2、3、4的下对角矩阵
[[0 0 0 0 0]
[1 0 0 0 0]
[0 2 0 0 0]
[0 0 3 0 0]
[0 0 0 4 0]]
使用 np.diag 方法：
In [1]: import numpy as np
In [2]: Z = np.diag(1+np.arange(4),k=-1)
...: print(Z)
[[0 0 0 0 0]
[1 0 0 0 0]
[0 2 0 0 0]
[0 0 3 0 0]
[0 0 0 4 0]]
其中，k 参数：大于0，表示与主对角线上移k，小于0下移k
