用字典理解替換兩個嵌套循環

Question

有沒有辦法用字典理解替換下面的代碼？

d = {} 

for i in xrange(A): 
    for j in xrange(B): 

     d[(i, j)] = f(i, j) 

我使用Python2.7，如果這有什麼區別。

Answer 1

d = {(i,j):f(i,j) for i in xrange(A) for j in xrange(B)} 

Answer 2

它通常是一個好主意，用itertools.product替換嵌套循環：

from itertools import product 
d = { p: f(*p) for p in product(range(A), range(B))}