用字典理解替换两个嵌套循环

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))}