如何求nXn矩阵向量的向量对角线？

def get_vector_diagonal(matrix):
    n = len(matrix)
    diagonal = []
    for i in range(n):
        diagonal.append(matrix[i][i])
    return diagonal

# 示例矩阵
matrix = [[1, 2, 3],
          [4, 5, 6],
          [7, 8, 9]]

vector_diagonal = get_vector_diagonal(matrix)
print(vector_diagonal)