Python怎么遍历文件夹获取文件名

K-seo • 2023-12-21 00:51 • 行业资讯 • 114 views

在Python中，我们可以使用os模块的os.walk()函数来遍历文件夹并获取文件名，os.walk()函数返回一个生成器，该生成器在每次迭代时产生一个三元组(root, dirs, files)，root是正在遍历的文件夹路径，dirs是一个包含root中所有子目录名称的列表，files是一个包含root中所有非目录子文件名称的列表。

以下是一个简单的示例，展示了如何使用os.walk()函数遍历文件夹并获取文件名：

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

在这个示例中，我们首先导入了os模块，我们定义了一个名为get_filenames的函数，该函数接受一个参数path，这是我们要遍历的文件夹的路径，在函数内部，我们使用for循环遍历os.walk()函数返回的生成器，对于每个生成的元素（即每个文件夹），我们打印出该文件夹中的所有文件名。

注意，os.path.join(root, file)用于将root和file连接成一个完整的路径，这是因为os.walk()函数返回的文件名只是相对于root的相对路径，而不是绝对路径。

我们还可以使用os.listdir()函数来获取文件夹中的文件名，os.listdir()函数返回指定文件夹包含的文件或文件夹的名字的列表，这个列表是按文件系统的顺序排序的，不依赖于文件系统中元素的位置。

以下是使用os.listdir()函数的示例：

import os
def get_filenames(path):
    for file in os.listdir(path):
        print(os.path.join(path, file))
get_filenames('/path/to/your/directory')

在这个示例中，我们没有使用os.walk()函数，而是直接使用os.listdir()函数获取文件夹中的所有文件名，我们使用os.path.join()函数将路径和文件名连接成一个完整的路径，并打印出来。

os.walk()函数和os.listdir()函数都是获取文件夹中文件名的有效方法，具体使用哪个函数取决于你的具体需求，如果你需要遍历文件夹的所有子目录和子文件，那么os.walk()函数可能是更好的选择，如果你只需要获取文件夹中的文件名，而不需要知道这些文件的具体位置，那么os.listdir()函数可能更适合你。