Все возможные перестановки ряда списков в Python
В Python у меня есть список списков n, каждого с переменным числом элементов. Как может я создавать единственный список, содержащий все возможные перестановки:
Например,
[ [ a, b, c], [d], [e, f] ]
Я хочу
[ [a, d, e] , [a, d, f], [b, d, e], [b, d, f], [c, d, e], [c, d, f] ]
Обратите внимание, что я не знаю n заранее. Я думал, что itertools.product будет правильным подходом, но он требует, чтобы я знал количество аргументов заранее
44
задан SilentGhost 17 May 2010 в 22:18
поделиться
3 ответа
Вам не нужно знать n заранее, чтобы использовать itertools.product
>>> import itertools
>>> s=[ [ 'a', 'b', 'c'], ['d'], ['e', 'f'] ]
>>> list(itertools.product(*s))
[('a', 'd', 'e'), ('a', 'd', 'f'), ('b', 'd', 'e'), ('b', 'd', 'f'), ('c', 'd', 'e'), ('c', 'd', 'f')]
77
ответ дан 26 November 2019 в 22:04
поделиться
Вы можете сделать это с помощью многоуровневого понимания списка:
>>> L1=['a','b','c']
>>> L2=['d']
>>> L3=['e','f']
>>> [[i,j,k] for i in L1 for j in L2 for k in L3]
[['a', 'd', 'e'], ['a', 'd', 'f'], ['b', 'd', 'e'], ['b', 'd', 'f'], ['c', 'd', 'e'], ['c', 'd', 'f']]
3
ответ дан 26 November 2019 в 22:04
поделиться
itertools.product работает для меня.
>>> l=[ [ 1, 2, 3], [4], [5, 6] ]
>>> list(itertools.product(*l))
[(1, 4, 5), (1, 4, 6), (2, 4, 5), (2, 4, 6), (3, 4, 5), (3, 4, 6)]
>>> l=[ [ 1, 2, 3], [4], [5, 6],[7,8] ]
>>> list(itertools.product(*l))
[(1, 4, 5, 7), (1, 4, 5, 8), (1, 4, 6, 7), (1, 4, 6, 8), (2, 4, 5, 7), (2, 4, 5, 8), (2, 4, 6, 7), (2, 4, 6, 8), (3, 4, 5, 7), (3, 4, 5, 8), (3, 4, 6,
7), (3, 4, 6, 8)]
>>>
4
ответ дан 26 November 2019 в 22:04
поделиться
Другие вопросы по тегам: