列表项的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]]