做一些汇总性的工作或者工作总结时,经常需要提取一批文件的文件名,以我为例,在完成一本图书的文稿工作后,最终需要统计这本书一共有哪些章、每章有多少页码,这时就需要将每章的文件名(不能包含文件扩展名)提取出来保存到一个文件中,以方便统计,本例就解决了我的这个问题。例如,《Python从入门到项目实践》这本书的所有文稿放在一个文件夹中,现在要统计这本书一共包含哪些章,并存放到一个文件中。
首先使用open()方法以写入模式打开文件,然后使用os.listdir()方法获取指定路径下的所有文件,并用os.path.splitext()方法对文件名和扩展名进行分离,获取到文件名,最后使用文件对象的write()方法将文件名写入文件中即可。代码如下:
import os # 导入os模块
with open('D:\Test.txt&