获取此矩阵的对角线

获取矩阵的对角线是指获取矩阵中从左上角到右下角的对角线上的元素。对角线可以分为主对角线和副对角线。

主对角线是指从矩阵的左上角到右下角的对角线，元素的行索引和列索引相等。副对角线是指从矩阵的右上角到左下角的对角线，元素的行索引和列索引之和等于矩阵的行数减一。

获取矩阵的对角线可以通过以下步骤实现：

遍历矩阵的行和列，当行索引等于列索引时，即为主对角线上的元素，将其添加到一个列表中。
遍历矩阵的行和列，当行索引加列索引等于矩阵的行数减一时，即为副对角线上的元素，将其添加到一个列表中。

以下是一个示例代码，用于获取矩阵的对角线：

def get_diagonal(matrix):
    diagonal = []
    rows = len(matrix)
    cols = len(matrix[0])
    
    # 获取主对角线上的元素
    for i in range(min(rows, cols)):
        diagonal.append(matrix[i][i])
    
    # 获取副对角线上的元素
    for i in range(min(rows, cols)):
        diagonal.append(matrix[i][cols-i-1])
    
    return diagonal

这是一个简单的示例代码，可以根据实际情况进行修改和优化。对角线在矩阵计算、图像处理、机器学习等领域中有广泛的应用。

腾讯云相关产品和产品介绍链接地址：