Question

我有一个值列表[1,2,3,4,5]，我想使用列表推导，numpy或其他任何工作返回所有可能连接的组合的列表。

通过连接，我的意思是列表中的项目对应该在组合中保持它们的邻接关系（不一定是与相邻值的整数彼此相邻，尽管那样可以。）

要清楚，就列表[[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],
...]

以下是不想要的项目类型：

{{1}}

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

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]]

Answer 2

from itertools import islice
def adj(l):
    ln = len(l)
    yield l
    for n in range(ln -1, 0, -1):
        yield from (list(islice(l, i, n + i)) for i in range(ln - n % ln+1))


l = [1, 2, 3, 4, 5]
print(list(adj(l)))
[[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]]

您也可以一次获取一个值，而无需将所有项目存储在内存中：

def adj(l):
    ln = len(l)
    yield l
    for n in range(ln - 1, 0, -1):
        yield from (islice(l, i, n + i) for i in range(ln - n % ln+1))


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

for ele in adj(l):
    print(list(ele))

输出：

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

或者在每个切片上调用一个函数：

for ele in adj(l):
    print(sum(ele))

输出：

Answer 3

我确信有很多方法可以做到这一点。我将建议一个简单的函数，它可能不是更加pythonic的解决方案。

def c_list(A):
    B = []
    for i in range(len(A)):
        for j in range(len(A) - i):
            B.append(A[j:j+i+1])
    return B

A = [ 1,2,3,4,5 ]
B = c_list(A)
print B

这个策略可以用于列表理解，如果这是你的事情：

C = [A[j:j+i+1] for i in range(len(A)) for j in range(len(A)-i)]

python连接列表项的组合

3 个答案: