列表项的python连接组合

Question

我有一个值列表 [1,2,3,4,5,...]，我想返回所有可能的连接组合的列表使用列表理解、numpy 或任何有效的方法。

通过连接，我的意思是列表中的项目对应该在组合中保持它们的邻接关系(不一定是具有相邻值的整数彼此相邻，尽管这样没关系)。

明确地说，这是我想要返回的列表[1,2,3,4,5]:

[[1,2,3,4,5],
[1,2,3,4],
[2,3,4,5],
[1,2,3],
[2,3,4],
[3,4,5],
[1,2],
[2,3],
[3,4],
[4,5],
[1],
[2],
[3],
[4],
[5]]

以下是我不想要的项目类型:

[[1,2,3,5],
[1,3,4,5],
...]

我看过很多关于获得所有可能组合的帖子，但我还没有看到有人提出有关具有连通性约束的组合的问题。有什么想法吗？

Answer 1

这是列表理解的一种方式:

>>> lst = [6, 7, 8, 9]
>>> [lst[i:j] for j in range(len(lst)+1) for i in range(j)]
[[6], [6, 7], [7], [6, 7, 8], [7, 8], [8], [6, 7, 8, 9], [7, 8, 9], [8, 9], [9]]

或者如果您重视顺序并且我从您的示例中猜对了:

>>> [lst[i:i+leng] for leng in range(len(lst), 0, -1) for i in range(len(lst)-leng+1)]
[[6, 7, 8, 9], [6, 7, 8], [7, 8, 9], [6, 7], [7, 8], [8, 9], [6], [7], [8], [9]]

或者使用itertools.combinations:

>>> [lst[i:j] for i, j in combinations(range(len(lst)+1), 2)]
[[6], [6, 7], [6, 7, 8], [6, 7, 8, 9], [7], [7, 8], [7, 8, 9], [8], [8, 9], [9]]

或者甚至根本没有任何列表理解(但现在我变得愚蠢了:-)

>>> list(map(lst.__getitem__, starmap(slice, combinations(range(len(lst)+1), 2))))
[[6], [6, 7], [6, 7, 8], [6, 7, 8, 9], [7], [7, 8], [7, 8, 9], [8], [8, 9], [9]]