python - 简单的列表理解

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