python列表理解;压缩列表?

您可以在单个列表理解中进行嵌套迭代:

[filename for path in dirs for filename in os.listdir(path)]

>>> listOfLists = [[1, 2],[3, 4, 5], [6]]
>>> reduce(list.__add__, listOfLists)
[1, 2, 3, 4, 5, 6]

我猜Itertools解决方案比这更有效,但这让人感觉非常恶心,并且避免了仅仅为了一个列表操作而导入一个库。

你可以在 itertools' recipes:

def flatten(listOfLists):
    return list(chain.from_iterable(listOfLists))

(注意:需要python 2.6+)

提出的问题 flatmap .提出了一些实现,但它们可能不需要创建中间列表。这里是一个基于迭代器的实现。

def flatmap(func, *iterable):
    return itertools.chain.from_iterable(map(func, *iterable))

In [148]: list(flatmap(os.listdir, ['c:/mfg','c:/Intel']))
Out[148]: ['SPEC.pdf', 'W7ADD64EN006.cdr', 'W7ADD64EN006.pdf', 'ExtremeGraphics', 'Logs']

在python 2.x中,使用 itertools.map 代替 map .

你可以直接做:

subs = []
for d in dirs:
    subs.extend(os.listdir(d))

可以使用普通加法运算符连接列表:

>>> [1, 2] + [3, 4]
[1, 2, 3, 4]

内置功能 sum 将按顺序添加数字,并可以选择从特定值开始:

>>> sum(xrange(10), 100)
145

将上面的内容合并以展开列表:

>>> sum([[1, 2], [3, 4]], [])
[1, 2, 3, 4]

现在您可以定义 flatmap :

>>> def flatmap(f, seq):
...   return sum([f(s) for s in seq], [])
... 
>>> flatmap(range, [1,2,3])
[0, 0, 1, 0, 1, 2]

编辑: 我刚刚在评论中看到了 another answer 我想这是正确的,python将不必要地构建和垃圾收集许多较小的列表与此解决方案。因此,最好的一点是,如果你习惯于函数式编程,它会非常简单和简洁。

import itertools
x=[['b11','b12'],['b21','b22'],['b31']]
y=list(itertools.chain(*x))
print y

Itertools将从python2.3及更高版本开始工作。

subs = []
map(subs.extend, (os.listdir(d) for d in dirs))

(但蚂蚁的回答更好,他要加1)

你可以试试 itertools.chain() ,像这样:

import itertools
import os
dirs = ["c:\\usr", "c:\\temp"]
subs = list(itertools.chain(*[os.listdir(d) for d in dirs]))
print subs

itertools.chain()。 返回迭代器,因此传递到 list() .

谷歌给我带来了下一个解决方案:

def flatten(l):
   if isinstance(l,list):
      return sum(map(flatten,l))
   else:
      return l

你可以使用 pyxtension :

from pyxtension.streams import stream
stream([ [1,2,3], [4,5], [], [6] ]).flatMap() == range(7)

def flat_list(arr):
    send_back = []
    for i in arr:
        if type(i) == list:
            send_back += flat_list(i)
        else:
            send_back.append(i)
    return send_back

If listA=[list1,list2,list3]
flattened_list=reduce(lambda x,y:x+y,listA)

这就行了。