获取列表中相邻元素的所有组合

Question

有没有办法获取所有相邻元素的组合呢？
这里有个例子：

补充说明：我想在字符串上使用，而不仅仅是数字。例如：[Explain,it,to,me,please]

列表：

[0,1,2,3,4]

结果：

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

比如说，结果中不会有 [0,2,3] 这样的组合，因为在上面的排序列表中，0 和 2 不是相邻的。

我试过用 itertools.combinations，但它会给出所有的组合。

Answer 1

就像cpburnz的回答一样，这里使用了嵌套的列表推导式来实现。
不过不同的是，这里的列表是按照长度从大到小的顺序生成的。

[L[j:i+j] for i in xrange(len(L), 0, -1) for j in xrange(len(L) - i + 1)]

不过，如果你只是想生成这些列表，可以考虑写一些更容易理解的代码，使用生成器：

def all_windows(L):
    for i in xrange(len(L), 0, -1):
        for j in xrange(len(L) - i + 1):
            yield L[j:i+j]

这段代码和上面的一样，但它是为了方便逐个遍历而设计的：

for sublist in all_windows(L):
    # does something with sublist

Answer 2

这里有一种方法，使用了来自itertools的一个旧的window配方。

from itertools import islice

def window(seq, n=2):
    "Returns a sliding window (of width n) over data from the iterable"
    "   s -> (s0,s1,...s[n-1]), (s1,s2,...,sn), ...                   "
    it = iter(seq)
    result = tuple(islice(it, n))
    if len(result) == n:
        yield result    
    for elem in it:
        result = result[1:] + (elem,)
        yield result

def all_windows(seq):
    for l in xrange(1, len(seq) + 1):
        for w in window(seq, l):
            yield w

Answer 3

你可以这样做：

>>> L = [0,1,2,3,4]
>>> result = [L[i:j] for i in xrange(len(L)) for j in xrange(i + 1, len(L) + 1)]
>>> pprint.pprint(result)
[[0],
 [0, 1],
 [0, 1, 2],
 [0, 1, 2, 3],
 [0, 1, 2, 3, 4],
 [1],
 [1, 2],
 [1, 2, 3],
 [1, 2, 3, 4],
 [2],
 [2, 3],
 [2, 3, 4],
 [3],
 [3, 4],
 [4]]

接下来，如果你想按照字符串的长度从长到短排序，同时再按照值从小到大排序，可以这样：

>>> result.sort(key=lambda x: (-len(x), x))
>>> pprint.pprint(result)
[[0, 1, 2, 3, 4],
 [0, 1, 2, 3],
 [1, 2, 3, 4],
 [0, 1, 2],
 [1, 2, 3],
 [2, 3, 4],
 [0, 1],
 [1, 2],
 [2, 3],
 [3, 4],
 [0],
 [1],
 [2],
 [3],
 [4]]

对于字符串，这样做会得到：

>>> L = ['Explain', 'it', 'to', 'me', 'please']
>>> result = [L[i:j] for i in xrange(len(L)) for j in xrange(i + 1, len(L) + 1)]
>>> result.sort(key=lambda x: (-len(x), x))
>>> pprint.pprint(result)
[['Explain', 'it', 'to', 'me', 'please'],
 ['Explain', 'it', 'to', 'me'],
 ['it', 'to', 'me', 'please'],
 ['Explain', 'it', 'to'],
 ['it', 'to', 'me'],
 ['to', 'me', 'please'],
 ['Explain', 'it'],
 ['it', 'to'],
 ['me', 'please'],
 ['to', 'me'],
 ['Explain'],
 ['it'],
 ['me'],
 ['please'],
 ['to']]