请用python帮我写一个还原数组的代码

以下是一个简单的还原数组的Python代码：

def restore_array(adjacent_pairs):
    # 创建一个字典，用于存储每个数字和相邻数字的关系
    num_dict = {}
    for pair in adjacent_pairs:
        num_dict[pair[0]] = num_dict.get(pair[0], []) + [pair[1]]
        num_dict[pair[1]] = num_dict.get(pair[1], []) + [pair[0]]
    
    # 找到第一个数字作为起点
    start_num = None
    for num in num_dict.keys():
        if len(num_dict[num]) == 1:
            start_num = num
            break
    
    # 从起点开始还原数组
    array = [start_num]
    visited = set()
    visited.add(start_num)
    while len(array) < len(num_dict):
        last_num = array[-1]
        next_num = num_dict[last_num][0] if last_num != num_dict[last_num][0] else num_dict[last_num][1]
        array.append(next_num)
        visited.add(next_num)
    
    return array

该函数接受一个二维列表`adjacent_pairs`，其中每个子列表表示数组中相邻的两个数字。函数通过遍历相邻数字的关系还原出原数组，并返回该数组。