リスト要素を並び替えてできる全ての組み合わせの作り方

実現したいこと

Python初心者です．リストの組み合わせの作成について質問させていただきます．

xk = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]

リストxkの要素を並び替え，１と０の並べ方の全ての組み合わせを示すリストcombinationを作りたいと考えています．

現在の方法

import itertools as irｑ

xk = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]

permutations = list(ir.permutations(xk))

combination = list(set(permutations))

この場合，1度，0や１を区別せず，順列を求めるので，数が膨大になってしまうため，メモリ不足となってしまいます．

###質問
上記リストxkを並び替えてできる792の要素を持つリストcombination
combination= [[0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1], [1, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1],・・・]

の作成過程についてご教示いただけますと幸いです。

行動規範の内容に同意します

回答2件

ベストアンサー

itertools.combinationsとnumpyを使うとこういう感じです。

lengthとzerosを変更して使ってください。

python
1>>> import itertools
2>>> import numpy as np
3>>>
4>>> length = 6
5>>> zeros = 3
6>>> ones = np.ones(length, dtype=int)
7>>>
8>>> for c in itertools.combinations(list(range(length)), zeros):
9...     result = ones.copy()
10...     np.subtract.at(result,list(c), 1)
11...     print(result)
12...
13[0 0 0 1 1 1]
14[0 0 1 0 1 1]
15[0 0 1 1 0 1]
16[0 0 1 1 1 0]
17[0 1 0 0 1 1]
18[0 1 0 1 0 1]
19[0 1 0 1 1 0]
20[0 1 1 0 0 1]
21[0 1 1 0 1 0]
22[0 1 1 1 0 0]
23[1 0 0 0 1 1]
24[1 0 0 1 0 1]
25[1 0 0 1 1 0]
26[1 0 1 0 0 1]
27[1 0 1 0 1 0]
28[1 0 1 1 0 0]
29[1 1 0 0 0 1]
30[1 1 0 0 1 0]
31[1 1 0 1 0 0]
32[1 1 1 0 0 0]

投稿2021/07/13 04:12

ppaul

総合スコア24670

J_RICE

2021/07/13 08:41

早速の回答，ありがとうございました。

行動規範の内容に同意します

Python
1a = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]
2
3def gen(n0, n1, i):   # n0: 0 の個数, n1: 1 の個数
4	if n0 == 0 and n1 == 0:
5		print(a)
6		return
7	if n0 > 0:
8		a[i] = 0
9		gen(n0-1, n1, i+1)
10	if n1 > 0:
11		a[i] = 1
12		gen(n0, n1-1, i+1)
13
14gen(5, 7, 0)

追記
リストcombination を作りたいのですね。

Python
1xk = [0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1]
2
3combination = []
4a = xk[:]
5
6def gen(n0, n1, i):   # n0: 0 の個数, n1: 1 の個数
7	if n0 == 0 and n1 == 0:
8		combination.append(a[:])
9		return
10	if n0 > 0:
11		a[i] = 0
12		gen(n0-1, n1, i+1)
13	if n1 > 0:
14		a[i] = 1
15		gen(n0, n1-1, i+1)
16
17gen(5, 7, 0)
18for e in combination:
19	print(e)