Python列表理解: populating each inner-list with two variables

Question

这适用于创建输出 [[0, 0], [1, 1], [2, 2], [0, 0], [1, 1], [2, 2], [0, 0 ], [1, 1], [2, 2]]:

[[k%3 for i in range(2)]for j in range(3)  for k in range(3)]

这适用于创建输出 [[0, 0], [0, 0], [0, 0], [1, 1], [1, 1], [1, 1], [2, 2 ], [2, 2], [2, 2]]:

[[j for i in range(2)]for j in range(3)  for k in range(3)]

现在，因为我的目标是获得如下所示的输出:[[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1 , 2], [2, 0], [2, 1], [2, 2]]

我尝试了以下操作，但出现错误:

[[j,k%3 for i in range(2)]for j in range(3)  for k in range(3)]

错误是:

[[j,k%3 for i in range(2)]for j in range(3) for k in range(3)]

语法错误:语法无效

Answer 1

您可以将其创建为元组:

[[(j,k%3) for i in range(2)]for j in range(3)  for k in range(3)]

输出如下:

>>> [[(j,k%3) for i in range(2)]for j in range(3)  for k in range(3)]
[[(0, 0), (0, 0)], [(0, 1), (0, 1)], [(0, 2), (0, 2)], [(1, 0), (1, 0)], [(1, 1), (1, 1)], [(1, 2), (1, 2)], [(2, 0), (2, 0)], [(2, 1), (2, 1)], [(2, 2), (2, 2)]]

---

编辑:

OP 发布的预期输出:

[[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2], [2, 0], [2, 1], [2, 2]

实现此目的的代码:

  [[j,k] for j in range(3)  for k in range(3)]