如何编辑排列

Question

每个数字都是一个端口。并且排列为我提供了所有可能的路线，但我的起始端口始终为 1，因此在所有组合中，我想要以 1 开头的组合。

这就是我到目前为止的全部内容

from itertools import permutations

routes = permutations([1 , 2, 3 ,4 , 5])
for i in list(routes):
      print(i)

有没有办法让我选择那些特定的组合？

Answer 1

您可以简单地在 2,3,4,5 的排列开头加一个 1。例如，

from itertools import permutations

routes = permutations([2, 3 ,4 , 5])
for i in routes:
      print([1,*i])

如果您真的想生成每个排列并过滤掉以 1 开头的排列，您可以执行以下操作。

from itertools import permutations

routes = permutations([1,2, 3 ,4 , 5])
for i in routes:
    if i[0] == 1:
        print(i)

how to edit a permutation