how to perform shortest path algorithm in python