在python中自动创建列表

Automatically creating a list in Python

本问题已经有最佳答案,请猛点这里访问。

我的代码:

1
2
3
4
5
red_images =  'DDtest/210red.png'
green_images = 'DDtest/183green.png'

b = [red_images, green_images]
shuffle(b)

我有几百张图片,为了使我的代码尽可能简洁(并扩展我的Python知识),我想知道我如何编写代码,自动将文件放在一个文件夹中,并将它们列为一个列表。

我在R中做过类似的事情,但我不知道如何在Python中做。


您也可以使用os

1
2
3
4
5
6
7
8
9
10
11
12
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)


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