numpy.
diag
提取对角线或构造对角线数组。
请参阅有关 numpy.diagonal 如果使用此函数提取对角线并希望写入结果数组,则返回副本还是视图取决于使用的numpy版本。
numpy.diagonal
如果 v 是二维数组,返回其 k 第四对角线。如果 v 是一维数组,返回带有 v 上 k 第四对角线。
有疑问的对角线。默认值为0。使用 k>0 主对角线上方的对角线,以及 k<0 主对角线以下的对角线。
提取的对角线或构造的对角线数组。
参见
diagonal
返回指定的对角线。
diagflat
创建一个二维数组,将扁平输入作为对角线。
trace
沿对角线求和。
triu
数组的上三角形。
tril
数组的下三角。
实例
>>> 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]])