简单列表理解

这两项工作:

[f for (subdir, dirs, files) in os.walk(rootdir) for f in files]

sum([files for (subdir, dirs, files) in os.walk(rootdir)], [])

$ find /tmp/test
/tmp/test
/tmp/test/subdir1
/tmp/test/subdir1/file1
/tmp/test/subdir2
/tmp/test/subdir2/file2
$ python
>>> import os
>>> rootdir = "/tmp/test"
>>> [f for (subdir, dirs, files) in os.walk(rootdir) for f in files]
['file1', 'file2']
>>> sum([files for (subdir, dirs, files) in os.walk(rootdir)], [])
['file1', 'file2']

files = [filename for (subdir, dirs, files) in os.walk(rootdir) for filename in files]

for (subdir, dirs, f) in os.walk(rootdir): files.extend(f)

import os, glob

files = [file for file in glob.glob('*') if os.path.isfile(file)]

如果文件有扩展名,则更简单的方法是:

import glob
files = glob.glob('*.*')