如何将多维列表最佳地转换为 Python 中的单个项目列表？ [复制]

Question

可能的重复：
Making a flat list out of list of lists in Python
Flatten (an irregular) list of lists in Python

输入一个列表如：[1,2,[3,4,[5,6,7,[8]]],[9],10]

期望的输出：[1,2,3,4,5,6,7,8,9,10]

目前我有以下递归函数：

def fix_list( li , l = [] ):
    for i in li:
        try:
            len( i )
            fix_list( i, l )
        except:
            l.append( i )
    return l

我觉得 try/except 捕获有点花哨和缓慢，我想知道是否有更好/更优化的方法来执行此任务。非常感谢所有输入。

Answer 1

这是一个迭代版本（最初是受 Artsiom Rudzenka 的启发），它使用切片分配来修改列表，而不是在每次通过时创建一个新列表。有趣的是，只需要一次通过列表！我对enumerate() 的使用有点不正统；我使用它来跟踪列表中的索引，该索引在我迭代它时可能会增长，但实际上并不使用实际的列表项。

def flattened(items, seqtypes=(list, tuple)):

    items = items[:]  # we will return a copy; remove to mutate original
    for i, _ in enumerate(items):
        while isinstance(items[i], seqtypes):
            items[i:i+1] = items[i]
    return items

print flattened([1,2,[3,4,[5,6,7,[8]]],[9],10])

生成器版本：

def flattener(items, seqtypes=(list, tuple)):

    items = items[:]
    for i, _ in enumerate(items):
        while isinstance(items[i], seqtypes):
            items[i:i+1] = items[i]
        yield items[i]

print list(flattener([1,2,[3,4,[5,6,7,[8]]],[9],10]))

这是一个list 子类，它有一个flatten() 方法。与sort() 和reverse() 方法一样，这会改变列表并返回None。

class listy(list):

    def flatten(self, seqtypes=(list, tuple)):
        for i, _ in enumerate(self):
            while isinstance(self[i], seqtypes):
                self[i:i+1] = self[i]

lst = listy([1,2,[3,4,[5,6,7,[8]]],[9],10])
lst.flatten()
print lst

编辑：我从原始版本中大大简化了这一点。但是，我看到问题已关闭。

Answer 2

另一种方法..

def flatten(old, new):                                                                                                                      
    for i in old:                                                                                                                           
        flatten(i, new) if isinstance(i, list) else new.append(i)                                                                           
    return new                                                                                                                              

if __name__ == '__main__':                                                                                                                  
    l1 = [1, 2, [3, 4, [5, 6, 7, [8]]], [9], 10]                                                                                            
    l2 = list()                                                                                                                             
    l3 = flatten(l1, l2)                                                                                                                     
    print l3

已简化，因此您不必向展平参数列表提供和清空列表..

def rec_flatten(old, new=list()):                                                                                                           
    for i in old:                                                                                                                           
        rec_flatten(i, new) if isinstance(i, list) else new.append(i)                                                                       
    return new                                                                                                                              

if __name__ == '__main__':                                                                                                                  
    l1 = [1, 2, [3, 4, [5, 6, 7, [8]]], [9], 10]                                                                                            
    x = rec_flatten(l1)                                                                                                                     
    print x

Answer 3

我已经尝试了以下解决方案并且它有效：

data = [1,2,[3,4,[5,6,7,[8]]],[9],10]

while any(isinstance(x, list) for x in data):
    tmpLst = []
    for x in data:
        if isinstance(x, list):
            tmpLst.extend(x)
        else:
            tmpLst.append(x)
    data = tmpLst

Answer 4

这是一个简单的生成器：

def flatten(nested):
    for item in nested:
        if isinstance(item, collections.Iterable) and not isinstance(item, basestring):
            for inner_item in flatten(item):
                yield inner_item
        else:
            yield item

不确定它是否“最佳”。

Answer 5

见this answer:

def flatten(l):
    for el in l:
        if isinstance(el, collections.Iterable) and not isinstance(el, basestring):
            for sub in flatten(el):
                yield sub
        else:
            yield el