Python OS 文件目录方法 os.walk()

Python OS 文件/目录方法 os.walk()

概述

os.walk() 方法用于通过在目录树中游走输出在目录中的文件名，向上或者向下。

os.walk() 方法是一个简单易用的文件、目录遍历器，可以帮助我们高效的处理文件、目录方面的事情。

在Unix，Windows中有效。

语法

**walk()**方法语法格式如下：

os.walk(top[, topdown=True[, onerror=None[, followlinks=False]]])

参数

• top – 是你所要遍历的目录的地址, 返回的是一个三元组(root,dirs,files)。
  • root 所指的是当前正在遍历的这个文件夹的本身的地址
  • dirs 是一个 list ，内容是该文件夹中所有的目录的名字(不包括子目录)
  • files 同样是 list , 内容是该文件夹中所有的文件(不包括子目录)
• topdown --可选，为 True，则优先遍历 top 目录，否则优先遍历 top 的子目录(默认为开启)。如果 topdown 参数为 True，walk 会遍历top文件夹，与top 文件夹中每一个子目录。
• onerror – 可选，需要一个 callable 对象，当 walk 需要异常时，会调用。
• followlinks – 可选，如果为 True，则会遍历目录下的快捷方式(linux 下是软连接 symbolic link )实际所指的目录(默认关闭)，如果为 False，则优先遍历 top 的子目录。

通俗解释*案例讲解

上面的官方解释是否看的有点云里雾里，没事，我们用案例讲解：

我们使用os.walk()方法进行对文件夹目录处理，获取到每个文件夹里的图片名称以及图片对应的标签。写入到txt文本中进行记录。

我们使用代码进行对大文件夹读取，

rootdata = r"D:/FreeDesk/大创项目/手写藏文字母识别/手写藏文字母数据/总数据/"
for root,dirs,files in os.walk(rootdata):
    print(root)
    print(dirs)
    print(files)

输出root：

输出dis：

输出files：

将每个图片信息，以及对应的标签写入txt文本中：

代码：

rootdata = r"D:/FreeDesk/大创项目/手写藏文字母识别/手写藏文字母数据/总数据/"
txt_list = [] #创建一个列表，里面存放  图片路径+‘\t’+图片标签
for root,dirs,files in os.walk(rootdata):
    # print(root)
    # print(dirs)
    # print(files)
    #拼接每个图片的绝对文件路径：
    for i in range(len(files)):
        # print(files[i])#输出的是每个图片的名称
        # print(root+"---"+files[i])  #shu输出每个每个图片的文件夹路径----图片名称
        # print(os.path.join(root, files[i]))  #拼接路径，

        # print(str(root).split("/")[-1])   #dui对root进行字符串切割，获得最后一个元素，代表每个图片的标签。
        class_flag = str(root).split("/")[-1]  #biaoqain标签
        data = os.path.join(root, files[i]) + '\t' + str(class_flag) + '\n'
        txt_list.append(data)

    with open('data.txt','w',encoding='UTF-8') as f:
        for txt  in txt_list:
            f.write(str(txt))

获得结果data.txt：

这样便使用了walk方法，对图片数据集进行简单的整理，方便深度学习时的数据集制作

如果觉得对您有一定的帮助，请点个小小关注哦，您的点赞关注是我创作的最大动力，努力的路上与君共勉！！！！

如果觉得对您有一定的帮助，请点个小小关注哦，您的点赞关注是我创作的最大动力，努力的路上与君共勉！！！！