获取目录中的文件 Python

在Python中，可以通过一些内置方法来获取目录中的文件列表。本文将介绍如何使用os.path和os模块来实现。

使用os模块

Python的os模块提供了获取目录中的文件列表的方法——os.listdir()。这个方法返回目录中的所有文件和文件夹的名称列表。

import os

directory = '/path/to/directory/'

file_list = os.listdir(directory)

print(file_list)

上面的代码将获取包含在目录'/path/to/directory/'中的文件和文件夹的名称，并将其打印出来。

处理文件列表可以使用Python的循环或列表推导式。例：

import os

directory = '/path/to/directory/'

file_list = os.listdir(directory)

for file_name in file_list:
    print(file_name)

或

import os

directory = '/path/to/directory/'

file_list = os.listdir(directory)

file_list = [file_name for file_name in file_list if file_name.endswith('.txt')]

print(file_list)

上面的代码演示了使用列表推导式来筛选以'.txt'结尾的文件。

使用os.path模块

os.path模块提供了更加精准的目录和文件数据。使用os.path.join()方法可安全地构建Windows和Unix路径。

import os

directory = '/path/to/directory/'

file_list = []

for root, dirs, files in os.walk(directory):
    for file_name in files:
        if file_name.endswith('.txt'):
            file_list.append(os.path.join(root, file_name))

print(file_list)

上面的代码演示了如何使用os.path模块来查找以'.txt'结尾的文件并将其存储到一个名为file_list的列表中。

总结：使用Python的os和os.path模块可轻松地获取目录中的文件列表。如果您需要对返回的文件列表进行筛选，则可以使用循环或列表推导式对文件列表进行处理。