在Python中关于使用os模块遍历目录的实现方法

使用os模块可以方便地遍历目录和操作文件，下面是在Python中使用os模块遍历目录的实现方法攻略：

导入os模块

import os

遍历目录

列出目录下的所有文件和目录

使用os.listdir()方法可以列出指定目录下的所有文件和目录。

def list_dir(path):
    """列出指定目录下的所有文件和目录"""
    for file_name in os.listdir(path):
        print(file_name)

列出指定类型的文件

使用os.listdir()和os.path.splitext()方法可以列出指定类型的文件，比如.txt。

def list_files(path, ext=".txt"):
    """列出指定目录下指定类型的文件"""
    for file_name in os.listdir(path):
        if os.path.splitext(file_name)[1] == ext:
            print(file_name)

示例说明

下面分别演示如何使用list_dir()和list_files()两个方法。

示例一：列出目录下的所有文件和目录

假设有如下目录结构：

root/
|-- dir1/
|   |-- file1.txt
|   |-- file2.jpg
|-- dir2/
|   |-- file3.txt
|   |-- file4.jpg
|-- file5.txt

代码如下：

list_dir("root")

输出结果如下：

dir1
dir2
file5.txt

示例二：列出指定类型的文件

假设有如下目录结构：

root/
|-- dir1/
|   |-- file1.txt
|   |-- file2.jpg
|-- dir2/
|   |-- file3.txt
|   |-- file4.jpg
|-- file5.txt

代码如下：

list_files("root", ".txt")

输出结果如下：

file5.txt
dir1/file1.txt
dir2/file3.txt