numpy—np.diag(v, k=0)

numpy.diag(v, k=0)

描述

提取对角线上的数组或者构造一个对角线数组

参数

v ： array_like
如果v是一个二维数组，则返回其与k对应的对角线上的数组。
如果v是一个一维数组，则返回一个二维数组，其中v位于与k对应的对角线上

k : int, optional
对角线。默认值为0。
使用k>0为主对角线以上的对角线，
k<0为主对角线以下的对角线

返回

提取的对角线上的数组或者构造的对角线数组

案例

cov1_fact = np.diag((1, 2, 3))

[[1 0 0]
 [0 2 0]
 [0 0 3]]