使用列表理解生成 2D 网格索引

Question

我想将3 x 3列表的行索引和列索引存储在列表中。它应该如下所示：

rc = [(0,0),(0,1),(0,2),(1,0),(1,1),(1,2),(2,0),(2,1),(2,2)]

如何使用 Python 中的列表理解来获取此列表？

Answer 1

有些人for认为列表理解中的多个循环是糟糕的风格。改用itertools.product()：

from itertools import product

list(product(range(3), repeat=2))

这输出：

[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]