python学习之文件处理

1. 加载识花数据集。接下来我们将 flower_photos 文件夹中的花朵图片都载入到进来，并且用图片所在的子文件夹作为标签值。

data_dir = 'flower_photos/'

contents = os.listdir(data_dir)

classes = [each for each in contents if os.path.isdir(data_dir + each)]

代码讲解：

#返回文件夹中 子文件及子文件夹 名称的列表，contents为['tulips', 'dandelion', 'LICENSE.txt', 'roses', 'sunflowers', 'daisy']
contents = os.listdir(data_dir)

#去掉子文件的名称，只保留子文件夹的名称。classes即为['tulips', 'dandelion', 'roses', 'sunflowers', 'daisy']
classes = [each for each in contents if os.path.isdir(data_dir + each)]