如何将此循环转换为列表解析？

Question

我正在尝试将以下代码转换为列表解析以提高紧凑性。谢谢！

grid = {}
for x in range(HOR_SQUARES):
    for y in range(VERT_SQUARES):
        grid[(x,y)] = GridSquare()

Answer 1

itertools.product很不错：

from itertools import product
grid = {tup: GridSquare() for tup in product(range(HOR_SQUARES), range(VERT_SQUARES))}

如果真的是方形（HOR_SQUARES == VERT_SQUARES），那么这更简单：

grid = {tup: GridSquare() for tup in product(range(HOR_SQUARES), repeat=2)}

Answer 2

尝试grid = dict(((x, y), GridSquare()) for x in range(HOR_SQUARES) for y in range(VERT_SQUARES))

Answer 3

您可以使用字典理解直接转到字典：

grid = {(x, y): GridSquare() for x in range(HOR_SQUARES) for y in range(VERT_SQUARES)}

紧凑性很好但是一旦你的理解开始变得越来越复杂，你可能会 也想格式化它们：

grid = {(x, y): GridSquare() 
        for x in range(HOR_SQUARES)
        for y in range(VERT_SQUARES)}