如何從固定集生成固定大小的所有不同二維數組？

Question

基本上我想要的是：給定一個 mxn 數組和一個非空集，生成包含這些元素的所有不同列表。 例如，給定一個2x2數組和集合{0,1,2} ，我希望返回：

[[0, 0], [0,0]]

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

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

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

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

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

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

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

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

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

等等。 我在 Python 中工作，但偽代碼或類似的東西會很好。 我似乎無法弄清楚如何做到這一點。

Answer 1

from itertools import product
x = [0, 1, 2] #define the set you want to work on
b = product(x, repeat=4)
# repeat should be the sum of dimension of the desired output 2+2 = 4 in this case
c = [[i[:2], i[2:]] for i in b]
#outputs [[(0, 0), (0, 0)], [(0, 0), (0, 1)], [(0, 0), (0, 2)], [(0, 0), (1, 0)], ...