python+os如何根据文件名自动生成文本

这篇文章将为大家详细讲解有关python+os如何根据文件名自动生成文本，小编觉得挺实用的，因此分享给大家做个参考，希望大家阅读完这篇文章后可以有所收获。

这可以进一步用来遍历处理数据

#! usr/bin/env python
# coding: utf-8

# 为了自动获取文件夹下的名字并将其写入文件，然后利用该文件来处理对应的文件


import os
import numpy as np
import cv2


# 先进行路径操作
currentRoot = os.getcwd() # 当前路径
imgRoot = currentRoot + '/img' # 图片所在的路径
os.chdir(imgRoot) # 切换到图片所在的路径


# 开始读取文件并将其名称写入一个文件
fileNames = open('imgName.txt','w') # 将要写入文件名的文件， w表示write
file = os.listdir(imgRoot) # 读取某个文件夹下的所有文件
for i in file: #采用文件句柄的方式遍历文件
 if os.path.splitext(i)[1]=='.jpg': # 判断是否是jpg文件，我们只对jpg文件进行操作
 data = str(i) + '\n' # 将文件名和换行放在一起，文件名采用相对路径
 fileNames.write(data) # 开始写入文件
fileNames.close() # 最后关掉文件

运行代码，可得到一个文件，可看下图

文件内容如下：

如果重新运行一次代码，发现文件内容不会变化。
其实这时候文件夹中已经有了imgName这个文件，但是其扩展名为txt，所以并不会被利用。

关于“python+os如何根据文件名自动生成文本”这篇文章就分享到这里了，希望以上内容可以对大家有一定的帮助，使各位可以学到更多知识，如果觉得文章不错，请把它分享出去让更多的人看到。