Python Shortest Walk 3D