One-liner从可能的目录列表中获取平面文件列表?

时间:2015-02-25 18:32:34

标签: python list python-2.7 directory

假设我们有一个目录列表,其中一些目录可能不存在

dirs = ['dir1','dir2','dir3']

为了论证,只有两个存在,其内容是:

dir1
  --file1a
  --file1b
dir2
  --file2a
  --file2b

获得所有文件的 flat 列表的最佳单行程序是什么?我最接近的是:

import os
files = [ os.listdir(x) for x in ['dir1','dir','dir3'] if os.path.isdir(x) ]

但是这给了我一个嵌套列表:

[['file1a','file1b'],['file2a','file2b']]

如果我想要['file1a', 'file1b', 'file2a', 'file2b'],我必须使用什么单线?

2 个答案:

答案 0 :(得分:4)

您可以将项目链接到一个列表中:

from itertools import chain as ch


files = list(ch.from_iterable(os.listdir(x) for x in ['dir1', 'dir', 'dir3'] if os.path.isdir(x)))

最短我能做到:

from itertools import chain as ch, ifilter as ifil, imap 
from os import path, listdir

files = list(ch.from_iterable(imap(listdir, ifil(path.isdir, ('dir1', 'dir', 'dir3')))))

如果您只想使用名称,那么只需遍历链对象。

答案 1 :(得分:2)

不确定为什么单线是您想要的,但在这里:

files = [path for dir_lst in map(os.listdir, filter(os.path.isdir, ['dir1','dir','dir3'])) for path in dir_lst]

首先过滤非目录(使用filter),然后map对剩下的os.listdir进行过滤,并使用两个for循环进行聚合。 / p>