教你使用Python的list_directory()函数快速获取目录列表

Python的os模块中提供了listdir()函数来获取指定目录的文件列表。该函数接受一个参数，即要获取文件列表的目录路径，并返回一个包含该目录下所有文件和文件夹的列表。

以下是一个示例代码，展示如何使用listdir()函数获取指定目录的文件列表：

import os

def list_directory(dir_path):
    try:
        # 使用listdir()函数获取目录下的文件列表
        file_list = os.listdir(dir_path)

        # 遍历文件列表并打印
        for file_name in file_list:
            print(file_name)

    except FileNotFoundError:
        print("指定的目录不存在")
            
# 调用list_directory函数，传入要获取文件列表的目录路径
list_directory("C:/Users/Username/Documents")

该代码示例中，首先导入了os模块。然后定义了一个名为list_directory()的函数，该函数接受一个参数dir_path，即要获取文件列表的目录路径。

在函数内部，首先使用listdir()函数获取目录下的文件列表，并将返回的结果赋值给变量file_list。然后使用for循环遍历file_list，并打印每个文件名。

如果指定的目录不存在，会捕获到FileNotFoundError异常，并打印出错误提示。

你可以根据自己的实际需求对这个示例代码进行修改和扩展，例如可以将文件列表返回出来供其他函数使用，或对文件名进行过滤等。

使用list_directory()函数可以帮助你更快速地获取目录下的文件列表，提高开发效率。