使用dict(python)中的特定键构建列表？

Question

我正在用Python实现Dijkstra搜索算法.在搜索结束时,我使用前驱映射重建最短路径,从目标节点的前任开始.例如:

path = []
path.append(destination)
previous = predecessor_map[destination]
while previous != origin:
    path.append(previous)
    previous = predecessor_map[previous]

有没有办法用更少的代码行(例如列表理解)来做到这一点？

Answer 1

我唯一的建议是摆脱轻微的代码重复:

path = []
previous = destination
while previous != origin:
    path.append(previous)
    previous = predecessor_map[previous]

除此之外,我认为您的代码实际上非常清晰,并且不太可能从任何缩短代码的尝试中受益.

最后,值得注意的是,上述内容也适用于destination == origin,而您的原始版本很可能不会(取决于predecessor_map填充的确切程度).不知道这是否与您的用例相关.