Python 目录包含文件

在Python中，目录包含文件是一种非常常见的操作，它可以让我们轻松地组织和管理我们的Python项目。本文将介绍如何使用Python语言操作目录中的文件。

检查目录是否存在

在Python中查找目录中的文件之前，我们需要先确保该目录存在。我们可以使用os.path模块中的isdir()函数来检查目录是否存在。

import os

if os.path.isdir('path/to/directory'):
    print("Directory exists")
else:
    print("Directory does not exist")

列出目录中的文件

一旦我们确定目录存在，我们就可以使用os模块中的listdir()函数来列出目录中的所有文件。这个函数返回目录中所有文件和子目录的名称列表。

import os

files = os.listdir('path/to/directory')
for file in files:
    print(file)

过滤文件

有时，我们只需要目录中的特定文件。我们可以使用endswith()方法来过滤文件名以特定扩展名结束的文件。

import os

files = os.listdir('path/to/directory')
for file in files:
    if file.endswith('.txt'):
        print(file)

递归遍历目录

有时候我们需要遍历整个目录树，包括所有子目录中的文件。为此，我们可以使用os模块的walk()函数。

import os

for root, dirs, files in os.walk('path/to/directory'):
    for file in files:
        print(os.path.join(root, file))

这将遍历整个目录树中的所有文件，并打印文件路径。

以上是Python操作目录中文件的常见操作，它们可以极大地帮助我们组织和管理我们的Python项目。