python - 如何从展平列表创建嵌套列表？

Question

我写了一个函数来创建一个嵌套列表。

例如:

input= ['a','b','c','','d','e','f','g','','d','s','d','a','']

我想在 '' 之前创建一个子列表

作为返回，我想要一个嵌套列表:

[['a','b','c'],['d','e','f','g'],['d','s','d','a']]

Answer 1

尝试下面的实现

>>> def foo(inlist, delim = ''):
    start = 0
    try:
        while True:
            stop = inlist.index(delim, start)
            yield inlist[start:stop]
            start = stop + 1
    except ValueError:
            # if '' may not be the end delimiter 
            if start < len(inlist):
                yield inlist[start:]
        return


>>> list(foo(inlist))
[['a', 'b', 'c'], ['d', 'e', 'f', 'g'], ['d', 's', 'd', 'a']]

另一种可能的实现方式是 itertools.groupby .但是你必须过滤结果以删除 ['']。但是，尽管它可能看起来是单行的，但上面的实现更像 pythonic，因为它直观且可读

>>> from itertools import ifilter, groupby
>>> list(ifilter(lambda e: '' not in e,
             (list(v) for k,v in groupby(inlist, key = lambda e:e == ''))))
[['a', 'b', 'c'], ['d', 'e', 'f', 'g'], ['d', 's', 'd', 'a']]