Python | numpy matrix.diagonal()

matrix.diagonal()是numpy模块中的一个函数，用于返回矩阵的对角线或者指定偏移量的斜线。

语法

下面是矩阵的对角线函数的语法：

numpy.matrix.diagonal()

参数

matrix.diagonal()函数的可选参数如下：

| 参数 | 描述 | | --------- | ------------------------------------------------------------ | | offset | 偏移量。对角线的偏移量为0。正数表示上对角线，负数表示下对角线。 |

返回值

返回一个一维数组，包含矩阵的对角线元素。

示例

下面的示例输入矩阵并使用matrix.diagonal()函数，返回矩阵的对角线。

import numpy as np

# 创建一个矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用 `matrix.diagonal()` 函数返回矩阵的对角线元素
print("Matrix Diagonal: ", matrix.diagonal())

输出结果：

Matrix Diagonal:  [1 5 9]

您还可以使用matrix.diagonal()函数返回矩阵的斜线。下面的示例显示如何使用offset参数返回矩阵的斜线：

import numpy as np

# 创建一个矩阵
matrix = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# 使用 `matrix.diagonal()` 函数返回矩阵的上对角线元素
print("Matrix Diagonal (Offset 1): ", matrix.diagonal(offset=1))

# 使用 `matrix.diagonal()` 函数返回矩阵的下对角线元素
print("Matrix Diagonal (Offset -1): ", matrix.diagonal(offset=-1))

输出结果：

Matrix Diagonal (Offset 1):  [2 6]
Matrix Diagonal (Offset -1):  [4 8]

总结

matrix.diagonal()函数用于返回矩阵的对角线或者指定偏移量的斜线。它是numpy模块中的一个非常有用的函数，可以用于各种类型的矩阵计算。