Automatically creating a list in Python

Question

My code:

red_images =  'DDtest/210red.png'
green_images = 'DDtest/183green.png'

b = [red_images, green_images]
shuffle(b)

I have several hundred pictures, and in hopes of making my code as concise as possible (and to expand my python knowledge), I was wondering how I write code that automatically takes the files in a folder and makes them a list.

I have done similar things in R, but I'm not sure how to do it in python.

You can look at [the Python documentation](https://docs.python.org/2/library/glob.html) on `glob` - it does what you want. — J Richard Snape, Feb 23 '15 at 17:08

score 3 · Answer 1 · 2015-02-23T17:15:31.067

3

You can also use os:

import os
from random import shuffle

# Base directory from which you want to search    
base_dir = "/path/to/whatever"

# Only take the files in the above directory
files = [f for f in os.listdir(base_dir) 
         if os.path.isfile(os.path.join(base_dir, f))]

# shuffle (in place)
shuffle(files)

edited Feb 23 '15 at 17:15

answered Feb 23 '15 at 17:08

score 1 · Accepted Answer · answered Feb 23 '15 at 17:07

1

import glob
my_new_list = glob.glob("/your/folder/*")

answered Feb 23 '15 at 17:07

RustyShackleford

25,262
6
22
38

So concise, I love it. Thank you. :) – y3trgfhsfgr Feb 23 '15 at 17:28

Automatically creating a list in Python

2 Answers2