在Python中自动创建列表

Question

我的代码：

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

b = [red_images, green_images]
shuffle(b)

我有几百张图片，为了让我的代码尽可能简洁（并扩展我的python知识），我想知道我是如何编写自动获取文件夹中的文件并使其成为列表的代码。

我在R中做过类似的事情，但我不确定如何在python中做到这一点。

Answer 1

您还可以使用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)

Answer 2

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