如何在python中展平嵌套的列表列表?

时间:2015-02-12 11:26:27

标签: python list reduce fold

我已经创建了一个函数来将路径拆分成python中的目录列表,如下所示:

splitAllPaths = lambda path: flatten([[splitAllPaths(start), end] if start else end for (start, end) in [os.path.split(path)]])

使用此辅助函数:

#these only work one directory deep
def flatten(list_of_lists):
    return list(itertools.chain.from_iterable(list_of_lists))

此函数的输出如下所示:

> splitAllPaths('./dirname/dirname2/foo.bar')
[[[['.'], 'dirname'], 'dirname2'], 'foo.bar']

现在我希望这是一个平面列表。我的尝试如下(带输出):

> flatten(splitAllPaths('./diname/dirname2/foo.bar'))
['.', 'd', 'i', 'r', 'n', 'a', 'm', 'e', 'd', 'i', 'r', 'n', 'a', 'm', 'e', '2', 'f', 'o', 'o', '.', 'b', 'a', 'r']

> reduce(list.__add__, (list(mi) for mi in splitAllPaths('./dirname/dirname2/foo.bar')))
me2/foo.bar')))
[[['.'], 'dirname'], 'dirname2', 'f', 'o', 'o', '.', 'b', 'a', 'r']

如何正确展开此列表(我也欢迎任何有关如何改进splitAllPaths功能的建议)?

3 个答案:

答案 0 :(得分:1)

这是一个不那么普遍的答案,但它解决了你原来的问题 - 虽然它的优雅是值得商榷的。

主要的想法是生成一个带有颠倒的列表(如['file', 'user', 'home', '/']顺序)非常简单,所以你可以创建它并最终反转它。所以归结为:

def split_paths(path):                                                       
    def split_paths_reverse(path):                                           
        head, tail = os.path.split(path)                                     
        while head and tail:                                                 
            yield tail                                                                                
            head, tail = os.path.split(head)                                 
        yield head                                                           

    return reversed(tuple(split_paths_reverse(path)))

示例:

test = '/home/user/file.txt'
print(list(split_paths(test)))

['/', 'home', 'user', 'file.txt']

您还可以通过将每个元素放入堆栈然后删除它来避免显式反转部分,但这取决于您。

答案 1 :(得分:1)

出现的Sortherst方式是:

listoflists = [[[['.'], 'dirname'], 'dirname2'], 'foo.bar']
str(listoflists).translate(None,"[]'").split(',')

答案 2 :(得分:0)

我通过编写(非一般)折叠器来解决这个问题。我认为@ L3viathan在评论中提供了更好,更实用的解决方案。

attempt = lambda list: attempt(list[0] + list[1:]) if len(list[0]) > 1 else list[0] + list[1:]

输出

> attempt([[[['.'], 'dirname'], 'dirname2'], 'foo.bar'])
['.', 'dirname', 'dirname2', 'foo.bar']

我现在也用一般foldr1

来写
> foldr1 = lambda func, list: foldr1(func, func(list[0], list[1:])) if len(list[0]) > 1 else func(list[0], list[1:])
> foldr1(list.__add__, [[[['.'], 'dirname'], 'dirname2'], 'foo.bar'])
['.', 'dirname', 'dirname2', 'foo.bar']

注意:比我更熟悉的人可以确认这是折叠而不是折叠(我经常让他们感到困惑)。