pythonの総当たりの組み合わせ

Question

pythonの総当たりの組み合わせについて質問させていただきます。

＜やりたいこと＞
リスト内の組み合わせの出力

 [1 2 3]

以下が理想出力になります。

[(1, 2), (1, 3), (2, 3), (1, 2, 3)]

実際に作成したコードが以下になります。

import itertools

b = [1 2 3]
n = len(n)
for i in range(2,n):
    b_pairs = (list(itertools.combinations(b,i)))
print(b_pairs)

このコードだと出力が[(1, 2), (1, 3), (2, 3)]となり、
liから2つ選ぶ組み合わせのみとなってしまい、3つ選ぶ組み合わせが
出力されないです。

分かる方がいましたら回答いただけると助かります。

※ご回答いただいた内容に質問させていただくこともあるかと思いますので、
※よろしければご返信いただければと思います。

Answer 1

bやnが唐突に湧いていますが、次のように解釈して回答します。

b = [1, 2, 3]
n = len(b)

勘違いしていそうな点
range(x, y) は x以上y未満の整数を送出します。

明らかなバグ
b_pairsを毎回上書きして、前の周回のデータを吹っ飛ばしてます。

---

こんなのを書きたかったのでは。何が違うか見比べてみて下さい。

import itertools

b = [1, 2, 3]
n = len(b)

b_pairs = []
for i in range(2, n+1):
    b_pairs += list(itertools.combinations(b, i))
    # 単に b_pairs += itertools.combinations(b, i) でも可
     
print(b_pairs)

実行結果 Wandbox

[(1, 2), (1, 3), (2, 3), (1, 2, 3)]

おまけ

こういうときはmore-itertoolsが便利です。

>>> list(more_itertools.powerset(b))
[(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
>>> [s for s in more_itertools.powerset(b)]
[(), (1,), (2,), (3,), (1, 2), (1, 3), (2, 3), (1, 2, 3)]
>>>
>>> [s for s in more_itertools.powerset(b) if len(s) >= 2]
[(1, 2), (1, 3), (2, 3), (1, 2, 3)]

Comments

[yone_yone]

ご回答ありがとうございます。
コードの修正をいたしました。
おまけまで提示していただき、とても参考になりました。
ベストアンサーとさせていただきます。

Answer 2

rangeの範囲が間違ってますね。 n -> n+1が正解です。

import itertools

li = [1, 2, 3]
n = len(li)
for i in range(2, n + 1):
    b_pairs = (list(itertools.combinations(li,i)))
    print(b_pairs)

あとこのコード、２ヶ所ほど変数の宣言をせずに使っている箇所があることが気になりました。

Comments

[yone_yone]

回答ありがとうございます。
コードの修正をいたしました。
rangeの範囲についてのご指摘ありがとうございます。