python批量读取文件名并写入txt文件中

下面是详细讲解“python批量读取文件名并写入txt文件中”的完整攻略。

1.问题描述

我们经常需要从某个文件夹中读取所有文件的名称，并将这些名称保存到一个txt文件中。在Python中，可以用os模块轻松实现此功能。

2.实现步骤

2.1引入OS模块

首先需要将os模块引入到你的脚本中，以便后面操作。

import os

2.2获取目标文件夹中所有文件的名称

使用os.listdir函数可以获取目标文件夹中所有文件的名称。

path = '/path/to/your/folder'
file_names = os.listdir(path)

其中，path为目标文件夹路径，file_names为所有文件名的列表。

2.3写入txt文件

使用open函数打开一个txt文件，并使用write函数将文件名写入txt文件。

with open('file_names.txt', 'w') as f:
    for file_name in file_names:
        f.write(file_name + '\n')

其中，file_names.txt为要写入的txt文件名，'w'表示打开文件进行写操作。此段代码循环遍历所有文件名，并将文件名写入txt文件中。

3.示例说明

下面是两个示例，分别介绍如何读取目标文件夹中的所有文件名，并将它们写入txt文件中。

示例一：

import os

path = '/path/to/your/folder'
file_names = os.listdir(path)

with open('file_names.txt', 'w') as f:
    for file_name in file_names:
        f.write(file_name + '\n')

示例二：

import os

path = '.'
file_names = [file for file in os.listdir(path) if file.endswith('.txt')]

with open('file_names.txt', 'w') as f:
    for file_name in file_names:
        f.write(file_name + '\n')

在示例二中，我们只选择以.txt结尾的文件进行处理，并将结果写入了同名的txt文件中。