dask.array.diag

dask.array.diag¶

dask.array.diag(v, k=0)[源代码]¶

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

此文档字符串是从 numpy.diag 复制的。

Dask 版本可能存在一些不一致性。

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

参数

varray_like: 如果 v 是一个二维数组，返回其第 k 条对角线的副本。如果 v 是一个一维数组，返回一个二维数组，其中 v 位于第 k 条对角线上。
kint, 可选: 问题中的对角线。默认值为0。使用 k>0 表示主对角线上方的对角线，使用 k<0 表示主对角线下方的对角线。

返回

参见

示例

>>> import numpy as np  
>>> 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]])

dask.array.delete

dask.array.diagonal