python 提取txt目录的代码

下面是 Python 代码示例，用于提取指定目录中所有 .txt 文件的文件名：

import os

directory = '/path/to/directory'
txt_files = [f for f in os.listdir(directory) if f.endswith('.txt')]
print(txt_files)

上面的代码使用了列表推导式来筛选出目录中所有以 .txt 结尾的文件。其中，os.listdir(directory)会返回一个列表，包含指定目录中的所有文件和文件夹的名称。