numpy diag 用法及代码示例

Lincoln Numpy 函数 4年前 0 153

用法：
numpy.diag(v, k=0)
提取对角线或构造对角行数组。

请参阅有关的更详细的文档numpy.diagonal如果使用此函数提取对角线并希望写入结果数组；返回的是副本还是视图取决于您使用的numpy版本。

–	描述
参数	v：： array_like 如果v是二维数组，则返回其k-th对角线的副本。如果v是一维数组，则返回M对角k-th上带有v的二维数组。 k：： int, 可选参数对角线问题。默认值为0。对于主对角线上方的对角线，使用k> 0，对于主对角线下方的对角线，使用k <0。
返回值	out：： ndarray 提取的对角线或构造的对角行数组。

–

描述

参数

v：： array_like

如果v是二维数组，则返回其k-th对角线的副本。如果v是一维数组，则返回M对角k-th上带有v的二维数组。
k：： int, 可选参数

对角线问题。默认值为0。对于主对角线上方的对角线，使用k> 0，对于主对角线下方的对角线，使用k <0。

返回值

out：： ndarray

提取的对角线或构造的对角行数组。

>>> x = np.arange(9).reshape((3,3))
>>> x
array([[0, 1, 2],
       [3, 4, 5],
       [6, 7, 8]])

>>> np.diag(x)
array([0, 4, 8])
>>> np.diag(x, k=1)
array([1, 5])
>>> np.diag(x, k=-1)
array([3, 7])

>>> np.diag(np.diag(x))
array([[0, 0, 0],
       [0, 4, 0],
       [0, 0, 8]])

声明：本站所有文章，如无特殊说明或标注，均为本站原创发布。任何个人或组织，在未征得本站同意时，禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益，可联系我们进行处理。