Простое понимание списка

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

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

Answer 3

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

Answer 4

import os, glob

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

если ваши файлы имеют расширения, то еще проще:

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