【pyhton】遍历文件夹

【python】遍历文件夹

文章目录

前言

os 模块提供了非常丰富的方法用来处理文件和目录。
这篇文章主要介绍如何运用os模块的方法遍历文件夹

一、os.walk()简介

概述

os.walk() 方法用于通过在目录树中游走输出在目录中的文件名。
os.walk() 方法是一个简单易用的文件、目录遍历器，可以帮助我们高效的处理文件、目录方面的事情。
该方法在Unix，Windows中有效。

语法

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 的子目录。

二、使用示例

1.遍历文件内的所有文件

代码如下（示例）：

import os

directory = r'D:\example\test'

for root, dirs, files in os.walk(directory):
    for file_name in files:
    	# 文件绝对路径
        file_path = os.path.join(root, file_name)
        print(file_path）
    for dir_name in dirs:
    	dir_path = os.path.join(root, dir_name)
        print(dir_path)

2.遍历获取txt文件

代码如下（示例）：

import os

directory = r'D:\ProgramData\iwyu\include-what-you-use\tests'

for root, dirs, files in os.walk(directory):
    for file_name in files:
        if file_name.endswith(".txt"):
            file_path = os.path.join(root, file_name)
            print(file_path)

示例使用endswith判断文件名（字符串）后缀是否为.txt。

总结

以上就是笔者要讲的内容，本文介绍了os,walk的简单用法和结合endswith的进阶用法，还有更多需求的同学欢迎在评论区留言。