I want a dictionary of files:
files = [files for (subdir, dirs, files) in os.walk(rootdir)] But I get,
files = [['filename1', 'filename2']] when I want
files = ['filename1', 'filename2'] How do I prevent looping through that tuple? Thanks!
- 3possible duplicate of Flattening a shallow list in pythonkennytm– kennytm2010-07-24 19:20:48 +00:00Commented Jul 24, 2010 at 19:20
- Also Making a flat list out of list of lists in pythonJohn Kugelman– John Kugelman2010-07-24 19:27:19 +00:00Commented Jul 24, 2010 at 19:27
- forget my files = [f for f in files for (subdir, dirs, files) in os.walk(rootdir)] I think that there are many side effects. I will delete the answerluc– luc2010-07-24 19:30:56 +00:00Commented Jul 24, 2010 at 19:30
Add a comment |
4 Answers
Both of these work:
[f for (subdir, dirs, files) in os.walk(rootdir) for f in files] sum([files for (subdir, dirs, files) in os.walk(rootdir)], []) Sample output:
$ 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'] 
Comments
for (subdir, dirs, f) in os.walk(rootdir): files.extend(f) 
1 Comment
Cristian Ciupitu
And before this
files = [].files = [filename for (subdir, dirs, files) in os.walk(rootdir) for filename in files] 
Comments
import os, glob files = [file for file in glob.glob('*') if os.path.isfile(file)] if your files have extensions, then even simpler:
import glob files = glob.glob('*.*') 