【Python学习笔记】os.walk

最新推荐文章于 2025-05-29 21:35:18 发布

原创最新推荐文章于 2025-05-29 21:35:18 发布 · 519 阅读

3 ·

CC 4.0 BY-SA版权

本文介绍如何使用Python的os模块遍历指定文件夹，批量获取图片的完整路径及对应的标签，并将这些信息写入data.txt文件。通过定义文件夹名称与标签的映射关系，实现对data文件夹下图片的自动化处理。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

使用os.walk函数可以一次得到data文件夹下所有图片的路径及其对应的标签
在这里插入图片描述


import os

# 定义文件夹名称和标签的对应关系
label_map = {
    'cat': 0,
    'dog': 1,
    'bat': 2
}

with open('data.txt', 'w') as f:

    # 遍历所有文件，root为当前文件夹，dirs是所有子文件夹名，files是所有文件名
    for root, dirs, files in os.walk('data'):
        for filename in files:
            # 获得文件完整路径
            filepath = os.path.join(root, filename)
            # 获取当前文件夹名称,即类别名
            dirname = os.path.split(root)[-1]
            # 得到类别名对应的标签
            label = label_map[dirname]
            # 将 图片路径+标签 存入data.txt文件中
            f.write('{},{}\n'.format(filepath, label))