Python 3.x 中如何使用os模块遍历目录中的文件

Python 3.x 中如何使用os模块遍历目录中的文件

在Python中，使用os模块可以很方便地对操作系统进行访问，其中包括遍历目录中的文件。遍历目录中的文件是数据科学领域中常见的任务，因为我们经常需要将一个目录中的所有文件读入内存进行处理或分析。本文将介绍如何使用Python中的os模块来遍历目录中的文件。

1. 使用os.listdir函数列出目录中的文件

os.listdir函数可以列出一个目录中的所有文件和子目录，返回一个包含文件名的列表。

import os
# 列出当前目录中的文件和子目录
files = os.listdir('.')
for file in files:
    print(file)

上述代码中，我们使用os.listdir('.')来列出当前目录中的文件和子目录，并遍历输出每个文件和子目录的名称。

2. 使用os.walk函数遍历目录中的所有文件

os.walk函数可以遍历一个目录及其子目录下的所有文件和子目录，返回一个包含三元组（dirpath, dirnames, filenames）的迭代器，其中：

dirpath：表示当前遍历的子目录的路径

dirnames：表示当前遍历的子目录中的子目录列表

filenames：表示当前遍历的子目录中的文件列表

import os
# 遍历当前目录及其子目录中的所有文件
for dirpath, dirnames, filenames in os.walk('.'):
    for filename in filenames:
        print(os.path.join(dirpath, filename))

上述代码中，我们使用os.walk('.')来遍历当前目录及其子目录中的所有文件，并打印出每个文件的完整路径。

3. 使用os.path.isfile函数判断一个路径是否为文件

os.path.isfile函数可以判断一个路径是否为文件。

import os
# 遍历当前目录及其子目录中的所有文件
for dirpath, dirnames, filenames in os.walk('.'):
    for filename in filenames:
        path = os.path.join(dirpath, filename)
        if os.path.isfile(path):
            print(path)

上述代码中，我们使用os.path.isfile函数判断遍历过程中的每个文件是否为文件，只有是文件才打印出其完整路径。

4. 使用os.path.isdir函数判断一个路径是否为目录

os.path.isdir函数可以判断一个路径是否为目录。

import os
# 遍历当前目录及其子目录中的所有目录
for dirpath, dirnames, filenames in os.walk('.'):
    for dirname in dirnames:
        path = os.path.join(dirpath, dirname)
        if os.path.isdir(path):
            print(path)

上述代码中，我们使用os.path.isdir函数判断遍历过程中的每个路径是否为目录，只有是目录才打印出其完整路径。

总结

本文介绍了如何使用Python中的os模块遍历目录中的文件。我们介绍了os.listdir函数和os.walk函数，以及使用os.path.isfile和os.path.isdir函数判断路径是否为文件或目录。这些函数可以很方便地对目录中的文件进行遍历和操作。