简单的列表理解

Question

我想要一个文件字典：

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

但是

files = [['filename1', 'filename2']] 

当我想要时

files = ['filename1', 'filename2']

我得到了如何防止循环遍历该元组？谢谢！

Answer 1

这两者都有效：

[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']

Answer 2

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

Answer 3

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

Answer 4

import os, glob

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

如果你的文件有扩展名，那就更简单了：

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