如何以这种方式对二维列表中的元素进行排序？

Question

我有一个像这样的二维列表（总是像正方形的）：

[[a1,a2,a3], [b1,b2,b3], [c1,c2,c3]]

我需要列出27个包含以下元素的列表：

[a1+b1+c1],[a1+b1+c2],[a1+b1+c3],......,[a3+b3+c2],[a3+b3+c3]

如果主列表始终由3个元素组成，我可以使用这种算法：

for i in list[0]

     for j in list[1]:

         for k in list[2]:

             some_func(i,j,k)

但是我的问题是主列表的长度每次都不同。有什么把戏吗？谢谢！

Answer 1

您可以使用product从itertools>>> from itertools import product>>> a =[[1,2,3],[4,5],[6,7,8,9]]>>> print ([i for i in product(*a)])[(1, 4, 6), (1, 4, 7), (1, 4, 8), (1, 4, 9), (1, 5, 6), (1, 5, 7), (1, 5, 8), (1, 5, 9), (2, 4, 6), (2, 4, 7), (2, 4, 8), (2, 4, 9), (2, 5, 6), (2, 5, 7), (2, 5, 8), (2, 5, 9), (3, 4, 6), (3, 4, 7), (3, 4, 8), (3, 4, 9), (3, 5, 6), (3, 5, 7), (3, 5, 8), (3, 5, 9)]>>> print ([sum(i) for i in product(*a)])[11, 12, 13, 14, 12, 13, 14, 15, 12, 13, 14, 15, 13, 14, 15, 16, 13, 14, 15, 16, 14, 15, 16, 17]通过采用每个子列表的笛卡尔乘积并对每个子列表求和来进行工作。当然，可以将其缩短为：>>> print (list(map(sum, product(*a))))[11, 12, 13, 14, 12, 13, 14, 15, 12, 13, 14, 15, 13, 14, 15, 16, 13, 14, 15, 16, 14, 15, 16, 17]使用map无视for循环。

Answer 2

您可以使用itertools.product>>> import itertools>>> s=[[1,2,3],[4,5,6],[7,8,9]]>>> list(map(sum, itertools.product(*s)))[12, 13, 14, 13, 14, 15, 14, 15, 16, 13, 14, 15, 14, 15, 16, 15, 16, 17, 14, 15, 16, 15, 16, 17, 16, 17, 18]