Python列表理解：用两个变量填充每个内部列表

Question

This works for creating the output [[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)]

This works for creating the output [[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)]

Now, since my goal was to get output that looks like this: [[0, 0], [0, 1], [0, 2], [1, 0], [1, 1], [1, 2], [2, 0], [2, 1], [2, 2]]

I tried the following, but I got an error:

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

The error is:

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

SyntaxError: invalid syntax

Answer 1

You can create it as a tuple:

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

Output will be like below:

>>> [[(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)]]

---

EDIT:

Expected Output Published by OP:

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

Code to achieve this:

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

Answer 2

if the expected result is

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

imo you simply need

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