Obtenir une liste fusionnée conditionnelle de listes disjointes en python

Question

Étant donné un ensemble de I = {0,1,2,3,4}, mon code produit des routes optimales réalisables et les résultats sont les suivants: x [0,2 ] = 1, x [2,4] = 1, x [4,3] = 1 (signifiant aller du nœud 0 au nœud 2 puis de 2 à 4, puis de 4 à 3) et restant x [i, j] = 0 (x [0,1] = 0, x [0,3] = 0, x [2,3] = 0, ...).

Ce que je dois une liste unique comme ceci: FinalRoute = [0,2,4,3] qui montre le même concept dans une seule liste.

J'ai essayé d'ajouter ces X où x [i, j] = 1 à FinalRoute mais je n'arrive pas à comprendre l'ordre des nœuds visités. Est-il possible de créer une liste unique (FinalRoute) en ajoutant x [i, j] s dans l'ordre visité?

Answer 1

Je ne sais pas ce que x [0, 2] est en python, mais en supposant que vous avez une liste 2 dimensions, où vous faites référence à des valeurs comme x [0] [2], non x [0, 2] et x [0] [2] == 1 signifie que vous allez de node0 à node2, voici une solution possible:

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)