列出所有连续子数组

22
我有一个整数数组[1,2,3],需要返回所有可能的连续子数组组合。

[[1],[2],[3],[1,2],[2,3],[1,2,3]]

如何使用Python处理呢?一种方法是使用两个循环和数组本身,但应该有更好的方法。


2
请展示您的解决方案,并说明其他可能更好的方式。 - Scott Hunter
1
@Wondercricket:虽然所有连续子列表确实是组合,但并非所有组合都是连续子列表。 - inspectorG4dget
5个回答

18

一行解决方案(我不知道"更好的方式"对你来说是什么意思)

L = [1,2,3]
[L[i:i+j] for i in range(0,len(L)) for j in range(1,len(L)-i+1)]

L=[1,2,3,4]
[L[i:i+j] for i in range(0,len(L)) for j in range(1,len(L)-i+1)]

你得到了,

[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]

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

10

简化检查员的解决方案:

def getAllWindows(L):
    for w in range(1, len(L)+1):
        for i in range(len(L)-w+1):
            yield L[i:i+w]

并且有一种不使用循环的解决方案:

def allSubArrays(L,L2=None):
    if L2==None:
        L2 = L[:-1]
    if L==[]:
        if L2==[]:
            return []
        return allSubArrays(L2,L2[:-1])
    return [L]+allSubArrays(L[1:],L2)

8
def kwindow(L, k):
    for i in range(len(L)-k+1):
        yield L[i:i+k]


def getAllWindows(L):
    for w in range(1, len(L)+1):
        yield from kwindow(L, w)

输出:

In [39]: for i in getAllWindows([1,2,3]): print(i)
[1]
[2]
[3]
[1, 2]
[2, 3]
[1, 2, 3]

6
一个基于itertools的方法:
import itertools

def allSubArrays(xs):
    n = len(xs)
    indices = list(range(n+1))
    for i,j in itertools.combinations(indices,2):
        yield xs[i:j]

例如:

>>> list(allSubArrays([1,2,3]))
[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]

0
li=[1,2,3]
l=[]
for i in range(length(li)):
    for j in range(i,len(li)+1):
        if i==j:                   *cancelling empty sublist item*
            continue
        else:
            subli=li[i:j]
            l.append(subli)
 print(l)

输出:

[[1], [1, 2], [1, 2, 3], [2], [2, 3], [3]]

请问您能否添加一段关于您代码的简短描述吗?链接:https://stackoverflow.com/help/how-to-answer - Carmoreno

网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接