python+os如何根據(jù)文件名自動(dòng)生成文本

這篇文章將為大家詳細(xì)講解有關(guān)python+os如何根據(jù)文件名自動(dòng)生成文本，小編覺得挺實(shí)用的，因此分享給大家做個(gè)參考，希望大家閱讀完這篇文章后可以有所收獲。

這可以進(jìn)一步用來遍歷處理數(shù)據(jù)

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

# 為了自動(dòng)獲取文件夾下的名字并將其寫入文件，然后利用該文件來處理對(duì)應(yīng)的文件


import os
import numpy as np
import cv2


# 先進(jìn)行路徑操作
currentRoot = os.getcwd() # 當(dāng)前路徑
imgRoot = currentRoot + '/img' # 圖片所在的路徑
os.chdir(imgRoot) # 切換到圖片所在的路徑


# 開始讀取文件并將其名稱寫入一個(gè)文件
fileNames = open('imgName.txt','w') # 將要寫入文件名的文件， w表示write
file = os.listdir(imgRoot) # 讀取某個(gè)文件夾下的所有文件
for i in file: #采用文件句柄的方式遍歷文件
 if os.path.splitext(i)[1]=='.jpg': # 判斷是否是jpg文件，我們只對(duì)jpg文件進(jìn)行操作
 data = str(i) + '\n' # 將文件名和換行放在一起，文件名采用相對(duì)路徑
 fileNames.write(data) # 開始寫入文件
fileNames.close() # 最后關(guān)掉文件

運(yùn)行代碼，可得到一個(gè)文件，可看下圖

文件內(nèi)容如下：

如果重新運(yùn)行一次代碼，發(fā)現(xiàn)文件內(nèi)容不會(huì)變化。
其實(shí)這時(shí)候文件夾中已經(jīng)有了imgName這個(gè)文件，但是其擴(kuò)展名為txt，所以并不會(huì)被利用。

關(guān)于“python+os如何根據(jù)文件名自動(dòng)生成文本”這篇文章就分享到這里了，希望以上內(nèi)容可以對(duì)大家有一定的幫助，使各位可以學(xué)到更多知識(shí)，如果覺得文章不錯(cuò)，請(qǐng)把它分享出去讓更多的人看到。