[py2rb] itertools.product

はじめに

移植やってます。
( from python 3.7 to ruby 2.7 )

product (Python)

import itertools

A = [('a', 'b', 'c'), ('d', 'e'), ('f', 'g')]

print(list(itertools.product(*A)))

# [('a', 'd', 'f'), ('a', 'd', 'g'), ('a', 'e', 'f'), ('a', 'e', 'g'), ('b', 'd', 'f'), ('b', 'd', 'g'), ('b', 'e', 'f'), ('b', 'e', 'g'), ('c', 'd', 'f'), ('c', 'd', 'g'), ('c', 'e', 'f'), ('c', 'e', 'g')]

ジェネレータ式の入れ子になった for ループとおよそ等価です。たとえば product(A, B) は ((x,y) for x in A for y in B) と同じものを返します。

ふむふむ。

product 失敗 (Ruby)

A = [['a', 'b', 'c'], ['d', 'e'], ['f', 'g']]

print(A.inject(:product))

# [[["a", "d"], "f"], [["a", "d"], "g"], [["a", "e"], "f"], [["a", "e"], "g"], [["b", "d"], "f"], [["b", "d"], "g"], [["b", "e"], "f"], [["b", "e"], "g"], [["c", "d"], "f"], [["c", "d"], "g"], [["c", "e"], "f"], [["c", "e"], "g"]] 

単に、injectすると、平坦化されていない状態で出力されます。

product + flatten 成功 (Ruby)

A = [['a', 'b', 'c'], ['d', 'e'], ['f', 'g']]

print(A.inject(:product).map(&:flatten))

# [["a", "d", "f"], ["a", "d", "g"], ["a", "e", "f"], ["a", "e", "g"], ["b", "d", "f"], ["b", "d", "g"], ["b", "e", "f"], ["b", "e", "g"], ["c", "d", "f"], ["c", "d", "g"], ["c", "e", "f"], ["c", "e", "g"]]

tupleではないのですが、同様の処理ができました。

product 応用問題 (Python)

import itertools

A = [(('H-', 'a'), 'b', 'c'), ('d', 'e'), ('f', 'g')]

print(list(itertools.product(*A)))

# [(('H-', 'a'), 'd', 'f'), (('H-', 'a'), 'd', 'g'), (('H-', 'a'), 'e', 'f'), (('H-', 'a'), 'e', 'g'), ('b', 'd', 'f'), 
('b', 'd', 'g'), ('b', 'e', 'f'), ('b', 'e', 'g'), ('c', 'd', 'f'), ('c', 'd', 'g'), ('c', 'e', 'f'), ('c', 'e', 'g')]

前回は、要素が一つずつでしたが、今回は要素数が異なります。

product 応用問題 (Ruby)

A = [[['H-', 'a'], 'b', 'c'], ['d', 'e'], ['f', 'g']]

print(A.inject(:product).map(&:flatten))

# [["H-", "a", "d", "f"], ["H-", "a", "d", "g"], ["H-", "a", "e", "f"], ["H-", "a", "e", "g"], ["b", "d", "f"], ["b", "d", "g"], ["b", "e", "f"], ["b", "e", "g"], ["c", "d", "f"], ["c", "d", "g"], ["c", "e", "f"], ["c", "e", "g"]]

print(A.inject(:product).map{ _1.flatten(1) }) # 1
print(A[0].product(*A[1..-1]))                 # 2

# [[["H-", "a"], "d", "f"], [["H-", "a"], "d", "g"], [["H-", "a"], "e", "f"], [["H-", "a"], "e", "g"], ["b", "d", "f"], ["b", "d", "g"], ["b", "e", "f"], ["b", "e", "g"], ["c", "d", "f"], ["c", "d", "g"], ["c", "e", "f"], ["c", "e", "g"]]

単なるflattenですと平坦化しすぎですので、深さを指定します。

2番目の書き方は、こちらを参照しました。

メモ

• Python の itertools.product を学習した
• 百里を行く者は九十里を半ばとす