在python中获取不相交列表的条件合并列表

Question

给定一组 I={0,1,2,3,4}，我的代码生成可行的最优路线，结果如下所示：x[0,2]=1 , x[2,4]=1, x[4,3]=1 （表示从节点 0 到节点 2，然后从 2 到 4，然后从 4到 3) 和剩余的 x[i,j]=0 ( x[0,1]=0 ,x[0,3]=0 , x[2,3]=0,.. .).

我需要一个单一列表，如下所示：FinalRoute=[0,2,4,3]，它在单一列表中显示相同的概念。 p>

我尝试将 x[i,j]=1 的 X 附加到 FinalRoute，但我无法确定访问节点的顺序。有没有办法通过按访问顺序附加 x[i,j]s 来创建单个列表（FinalRoute）？

Answer 1

不确定 python 中的 x[0, 2] 是什么，但假设你有一个二维列表， 您将值引用为 x[0][2]，而不是 x[0, 2] 和 x[0][2] == 1 表示您从 node0 转到 node2，这是一个可能的解决方案：

matrix = [[0, 0, 1, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 1], [0, 0, 0, 0, 0], [0, 0, 0, 1, 0]]

path = []
node = 0
while node is not None:
    path.append(node)
    node = matrix[node].index(1) if 1 in matrix[node] else None

print(path)