itertools productを複数回適用した際の仕様

Question

itertools productによる直積を複数の要素に対して適用する際に、
要素a,b,c,dを一回でproductする結果と
要素a,b,cを一度productし、その結果(p_a_b_c)と要素dをproductする結果は異なります。

この後者を前者と同じ結果にする方法はありますか？

import itertools

a = (1,2)
b = (11,12)
c = (101,102)
d = (1001,1002) 

product_ = tuple(itertools.product(a,b,c,d))
>>>
((1, 11, 101, 1001), (1, 11, 101, 1002), (1, 11, 102, 1001), (1, 11, 102, 1002), (1, 12, 101, 1001), (1, 12, 101, 1002), (1, 12, 102, 1001), (1, 12, 102, 1002), (2, 11, 101, 1001), (2, 11, 101, 1002), (2, 11, 102, 1001), (2, 11, 102, 1002), (2, 12, 101, 1001), (2, 12, 101, 1002), (2, 12, 102, 1001), (2, 12, 102, 1002))

p_a_b_c= tuple(itertools.product(a,b,c))
product_2 = tuple(itertools.product(p_a_b_c, d))
>>>
(((1, 11, 101), 1001), ((1, 11, 101), 1002), ((1, 11, 102), 1001), ((1, 11, 102), 1002), ((1, 12, 101), 1001), ((1, 12, 101), 1002), ((1, 12, 102), 1001), ((1, 12, 102), 1002), ((2, 11, 101), 1001), ((2, 11, 101), 1002), ((2, 11, 102), 1001), ((2, 11, 102), 1002), ((2, 12, 101), 1001), ((2, 12, 101), 1002), ((2, 12, 102), 1001), ((2, 12, 102), 1002))

Comments

[tiitoi]

質問が重複してるので、古いほうの質問は削除依頼してください
https://teratail.com/questions/279077

[jckkvs]

大丈夫です。

Answer 1

product_2 = tuple((*x, y) for x, y in itertools.product(p_a_b_c, d))

です。

実行結果

>>> product_2 = tuple((*x, y) for x, y in itertools.product(p_a_b_c, d))
>>> print(product_2)
((1, 11, 101, 1001), (1, 11, 101, 1002), (1, 11, 102, 1001), (1, 11, 102, 1002), (1, 12, 101, 1001), (1, 12, 101, 1002), (1, 12, 102, 1001), (1, 12, 102, 1002), (2, 11, 101, 1001), (2, 11, 101, 1002), (2, 11, 102, 1001), (2, 11, 102, 1002), (2, 12, 101, 1001), (2, 12, 101, 1002), (2, 12, 102, 1001), (2, 12, 102, 1002))

Answer 2

公式ドキュメントに載っている実装例の一部を変更すれば、実現できます。

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

def product(*args, repeat=1):
    pools = [tuple(pool) for pool in args] * repeat

    result = [[]]
    for pool in pools:
        result = [x + list(y) if isinstance(y, tuple) else x + [y] for x in result for y in pool]

    for prod in result:
        yield tuple(prod)


A = range(2)
B = range(3)
C = range(2)

for x in product(product(A, B), C):
    print(x)
# (0, 0, 0)
# (0, 0, 1)
# (0, 1, 0)
# (0, 1, 1)
# (0, 2, 0)
# (0, 2, 1)
# (1, 0, 0)
# (1, 0, 1)
# (1, 1, 0)
# (1, 1, 1)
# (1, 2, 0)
# (1, 2, 1)

import itertools

for a, b in zip(product(product(A, B), C), itertools.product(A, B, C)):
    assert a == b  # 一致することを確認

変更箇所

tuple の場合は list にしてから結合する

- result = [x+[y] for x in result for y in pool]
+ result = [x + list(y) if isinstance(y, tuple) else x + [y] for x in result for y in pool]