Python实现读取目录所有文件的文件名并保存到txt文件代码

下面是Python实现读取目录所有文件的文件名并保存到txt文件代码的完整实例教程：

1. 导入必要的库

import os

2. 定义函数获取目录下所有文件名

def get_files_name(dir_path):
    file_name_list = []
    for root, dirs, files in os.walk(dir_path):
        for f in files:
            file_name_list.append(f)
    return file_name_list

该函数接收一个目录路径参数，返回该目录下所有文件的文件名列表。其中使用了os.walk函数遍历目录下所有文件，然后将文件名添加到列表中。

3. 调用函数并将列表保存为txt文件

if __name__ == '__main__':
    dir_path = 'my_dir_path'
    file_name_list = get_files_name(dir_path)
    with open('file_names.txt', 'w') as f:
        for name in file_name_list:
            f.write(name + '\n')

在主函数中调用get_files_name函数获取目录下所有文件的文件名列表，并将列表保存为txt文件。在这里，我们使用了with关键字打开文件，这样可以保证文件在写入完成后会被正常关闭，防止出现文件无法关闭的问题。

示例说明

示例1

如果目录结构如下：

my_dir_path/
    file1.txt
    file2.docx
    folder1/
        file3.jpg
    folder2/
        file4.xlsx

则调用get_files_name函数返回的列表为：

['file1.txt', 'file2.docx', 'file3.jpg', 'file4.xlsx']

在调用with关键字将列表保存为txt文件后，txt文件的内容为：

file1.txt
file2.docx
file3.jpg
file4.xlsx

示例2

如果目录结构如下：

my_dir_path/
    empty_folder1/
    empty_folder2/

则调用get_files_name函数返回的列表为空，因为该目录下没有文件。

在调用with关键字将列表保存为txt文件后，txt文件的内容为空。