获取目录中的文件列表

在Python中，我们可以使用os模块的listdir()函数来获取目录中的所有文件列表。listdir()函数将返回包含目录中所有文件和目录的列表。

import os

def get_file_list(path):
    """
    获取指定目录下的所有文件列表
    :param path: 指定目录的路径
    :return: 文件列表
    """
    return os.listdir(path)

该函数接受一个字符串参数path作为目录的路径，并返回该目录下的所有文件和目录的列表。

示例用法

path = "/Users/username/Documents"
file_list = get_file_list(path)
print(file_list)

输出结果为：

['study_notes.txt', 'resume.docx', 'pictures', 'projects', 'invoices']

遍历目录中所有文件

如果我们需要遍历目录中的所有文件（包括子目录中的文件），可以使用os.walk()函数，它将返回一个生成器，每次产生三个值：父目录路径、所有子目录的列表和当前目录中所有文件的列表。

import os

def traverse_files(path):
    """
    遍历指定目录下的所有文件
    :param path: 指定目录的路径
    """
    for root, dirs, files in os.walk(path):
        for file in files:
            file_path = os.path.join(root, file)
            print(file_path)

该函数接受一个字符串参数path作为目录的路径，并遍历目录中的所有文件（包括子目录中的文件），并在控制台输出每个文件的完整路径。

示例用法

path = "/Users/username/Documents"
traverse_files(path)

输出结果为：

/Users/username/Documents/study_notes.txt
/Users/username/Documents/resume.docx
/Users/username/Documents/pictures/flower.jpg
/Users/username/Documents/projects/project1/file1.py
/Users/username/Documents/invoices/invoice1.pdf

以上就是如何在Python中获取目录中的文件列表以及遍历目录中的所有文件的介绍。