Python ある範囲の整数を組で取り出すループを作りたい

Question

for (k,l,n) in [(-2,-2,-2),(-1,-2,-2),(-2,-1,-2),.....(2,2,1),(2,2,2)]:

上のように、k,l,nがそれぞれ-2～2までの整数値を取るとき、[]内を上のようにではなく、簡潔に書きたいです。
そのような書き方を教えていただきたいです。些末な質問で申し訳ありませんがよろしくお願いいたします

Answer 1

こんな方法も。
pats.py

PATS = [-2, -1, 0, 1, 2]
BASE = len(PATS)

for x in range(0, BASE ** 3):
    i, j, k = PATS[x // (BASE ** 2)], PATS[(x // BASE) % BASE], PATS[x % BASE]
    print(i, j, k)

実行例

途中省略

Comments

[Reikirin]

回答ありがとうございます。なるほど、こんなやり方もあるんですね。
すみません、気になった部分なのですが、
PATS[x // (BASE ** 2)], PATS[(x // BASE) % BASE], PATS[x % BASE]
の部分が理解できなかったので、もう少し詳しく教えていただけると幸いです。

[katoy]

1..999 の整数を 3 桁に数字の文字列にするにはどうしますか、
1 の位は n % 10、10 の 位は (n / 10) % 10, 100 の位は n / 100 です。

そして、各位には 0,,, 9 が現れて,全ての組み合わせが現れます。
これは 10 進数の場合です。

この質問の回答では、 5 進数として扱い、 0..4 の数字の代わりに -2, -1, 0 ,1, 2 を割り当てるようにして、
-2, -1, 0 ,1 ,2 の３つ組 の茎合わせがすべて現れます。

[Reikirin]

なるほど、％の理解が甘く、わかりませんでした。％の処理では、小数点以下を切り捨てた、割り算＋ガウス記号的な処理になるのですね。理解いたしました。

Answer 2

3重のforループを使うとコンパクトなコードになります。

for i in range(-2,3):
    for j in range(-2,3):
        for k in range(-2,3):
            print((i,j,k))

-2～2までの整数値をrange関数で作る際は、range(-2,2)ではなくrange(-2,3)であることに注意。

Comments

[Reikirin]

ありがとうございます。参考にさせて頂きます。

Answer 3

A = [-2, -1, 0, 1, 2] としたとき、すべての a ∈ A^3 を取得するということなので、直積の要素を生成する itertools.product をお使いください。

itertools --- 効率的なループ実行のためのイテレータ生成関数 — Python 3.8.2rc1 ドキュメント

from itertools import product

lst = [-2, -1, 0, 1, 2]  # range(-2, 3) でもよい

for k, l, n in product(lst, repeat=3):
    print(k, l, n)

Comments

[Reikirin]

まさにやりたかったことがそれです。前回に引き続きありがとうございました。