精梳for循環

Question

當壓扁的循環，你會做這樣的事情：

for a, b, c in itertools.product(x1, x2, x3): 
    ... 

但是，你會如何處理的情況下的要素之一是使用作爲參數來檢索列表也被遍歷？例如：

for a, b in itertools.product(x1, get_b_elements(a)): 
    ... 

這甚至有可能嗎？

Answer 1

for a in x1: 
    for b in get_b_elements(a): 
     #do something with (a, b) 

只是列出一種替代方案：

for (a, b) in [(a_i, b_i) for a_i in x1 for b_i in get_b_elements(a_i)]: 
    #do something with (a, b) 

由於@wim筆記，沒有這些 「扁平化」 的循環，只要你想。

Answer 2

試試這個：

alphabets = [a,b,c,d] 
xs = [x1,x2,x3] 

itertools.product(alphabets, xs)