Как генерировать списковые комбинации?

python combinations

1928 просмотра

4 ответа

3060 Репутация автора

Я хочу составить список списков, который представляет все возможные комбинации чисел 0 и 1. Списки имеют длину n.

Вывод должен выглядеть следующим образом. Для n = 1:

[ [0], [1] ]

Для n = 2:

[ [0,0], [0, 1], [1,0], [1, 1] ]

Для n = 3:

[ [0,0,0], [0, 0, 1], [0, 1, 1]... [1, 1, 1] ]

Я посмотрел на itertools.combinsk, но это создает кортежи, а не списки. [0,1] и [1,0] являются различными комбинациями, тогда как есть только один кортеж (0,1) (порядок не имеет значения).

Любые намеки или предложения? Я пробовал некоторые рекурсивные методы, но я не нашел решения.

Автор: Nate Reed Источник Размещён: 28.07.2013 04:32

Ответы (4)


4 плюса

22642 Репутация автора

Решение

Ты ищешь itertools.product(...).

>>> from itertools import product
>>> list(product([1, 0], repeat=2))
[(1, 1), (1, 0), (0, 1), (0, 0)]

Если вы хотите преобразовать внутренние элементы в listтип, используйте понимание списка

>>> [list(elem) for elem in product([1, 0], repeat =2)]
[[1, 1], [1, 0], [0, 1], [0, 0]]

Или с помощью map()

>>> map(list, product([1, 0], repeat=2))
[[1, 1], [1, 0], [0, 1], [0, 0]]
Автор: Sukrit Kalra Размещён: 28.07.2013 04:33

2 плюса

7252 Репутация автора

Используйте itertools.product, присваивая repeatn.

from itertools import product
list(product([0,1], repeat=n))

Демо-версия:

>>> list(product([0,1], repeat=2))
[(0, 0), (0, 1), (1, 0), (1, 1)]
>>> list(product([0,1], repeat=3))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]
Автор: zhangyangyu Размещён: 28.07.2013 04:34

2 плюса

176191 Репутация автора

>>> from itertools import product
>>> list(product([0, 1], repeat=2))
[(0, 0), (0, 1), (1, 0), (1, 1)]
>>> 
>>> list(product([0, 1], repeat=3))
[(0, 0, 0), (0, 0, 1), (0, 1, 0), (0, 1, 1), (1, 0, 0), (1, 0, 1), (1, 1, 0), (1, 1, 1)]

Чтобы получить список списка, вы можете сделать:

>>> map(list, list(product([0, 1], repeat=2)))
[[0, 0], [0, 1], [1, 0], [1, 1]]
Автор: Rohit Jain Размещён: 28.07.2013 04:35

1 плюс

392561 Репутация автора

Просто чтобы добавить немного разнообразия, вот еще один способ достижения этого:

>>> [map(int, format(i, "03b")) for i in range(8)]
[[0, 0, 0], [0, 0, 1], [0, 1, 0], [0, 1, 1], [1, 0, 0], [1, 0, 1], [1, 1, 0], [1, 1, 1]]
Автор: Sven Marnach Размещён: 28.07.2013 05:53
Вопросы из категории :
32x32