将对角线上的元素移位到行首

def move_diagonal_to_front(matrix):
    rows = len(matrix)
    cols = len(matrix[0])
    result = []

    # 将对角线上的元素移位到行首
    for i in range(rows):
        row = []
        for j in range(cols):
            if i == j:
                row.append(matrix[i][j])
        for j in range(cols):
            if i != j:
                row.append(matrix[i][j])
        result.append(row)

    return result