【Python学习笔记】os.walk

本文介绍如何使用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))

图片路径,标签
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值