combina_trajetorias.py

import unittest
import itertools

class CombinaTrajetoriasTests(unittest.TestCase):
    def test_combina_3_trajetorias(self):
        trajetorias = (801, 802, 803)
        trajetorias_combinadas = {801:[(801,802),(801,803)],
                                  802:[(802,801),(802,803)],
                                  803:[(803,801),(803,802)]
                                 }
        self.assertEqual(combina_trajetorias(trajetorias), trajetorias_combinadas)

    def test_combina_4_trajetorias(self):
        trajetorias = (801, 802, 803, 804)
        trajetorias_combinadas = {801:[(801,802),(801,803), (801,804)],
                                  802:[(802,801),(802,803), (802,804)], 
                                  803:[(803,801),(803,802), (803,804)],
                                  804:[(804,801),(804,802), (804,803)]
                                 }
        self.assertEqual(combina_trajetorias(trajetorias), trajetorias_combinadas)

def combina_trajetorias(trajetorias):
    trajetorias_combinadas = {}
    for trajetoria in trajetorias:
        trajetorias_combinadas[trajetoria] = list(itertools.product([t for t in trajetorias if t == trajetoria], [t for t in trajetorias if t != trajetoria]))
    return trajetorias_combinadas

if __name__ == '__main__':
    unittest.main()

#Outra tentativa foi usar permutação, mas não ele traz todas as combinações juntas:
#trajetorias = (801, 802, 803)
#list(itertools.permutations(trajetorias, 2))
#[(801, 802), (801, 803), (802, 801), (802, 803), (803, 801), (803, 802)]
#trajetorias = (801, 802, 803, 804)
#list(itertools.permutations(trajetorias, 2))
#[(801, 802), (801, 803), (801, 804), 
 #(802, 801), (802, 803), (802, 804),
 #(803, 801), (803, 802), (803, 804),
 #(804, 801), (804, 802), (804, 803)]