快捷導(dǎo)航

Python3.遍歷某文件夾提取特定文件名的實(shí)例

更新時(shí)間：2018年04月26日 14:33:59 作者：modabao

下面小編就為大家分享一篇Python3.遍歷某文件夾提取特定文件名的實(shí)例，具有很好的參考價(jià)值，希望對大家有所幫助。一起跟隨小編過來看看吧

批量處理文件時(shí)，常需要先遍歷某個(gè)路徑提取特定條件的文件名。這篇寫一個(gè)暴力遍歷但很簡潔的方法，真的非常簡潔但是非常暴力。

例子目標(biāo)是：獲得存放遙感數(shù)據(jù)的文件夾下文件夾名以“_BAD”結(jié)尾的文件夾內(nèi)容。因?yàn)樵撐募路旨壓芏啵?月/產(chǎn)品類型/），目標(biāo)文件夾很多且存在在最后一級，手動(dòng)查看很煩。

代碼如下（知識(shí)點(diǎn)總結(jié)代碼后）：

# -*- coding: utf-8 -*-
"""
遍歷某路徑下所有文件夾，獲得特定文件夾下所有文件
很暴力，真的遍歷了所有的文件夾
20180124
@author: 墨大寶
"""
import os
TARGETPATH = r'F:\MODIS_DATA'
records = []
for currentDir, _, includedFiles in os.walk(TARGETPATH):
 if not currentDir.endswith('_BAD'): continue
 else:
  records.append(currentDir) # 將以“_BAD”結(jié)尾的文件夾名加入records
  records.extend(includedFiles) # 將該文件夾內(nèi)的文件名列表擴(kuò)展到records
# 將records寫入.txt
txtFile = open(os.path.join(TARGETPATH, '02_04_BAD.txt'), 'w')
txtFile.write(os.linesep.join(records))
txtFile.close()
# 將排序后的records寫入.txt
with open(os.path.join(TARGETPATH, '02_04_BAD_SORTED.txt'), 'w') as txtFile:
 txtFile.write('\n'.join(sorted(records)))

os.walk(）返回Directory tree generator。每次生成格式為（dirpath, dirnames, filenames）的tuple，元素依次是當(dāng)前路徑、當(dāng)前路徑下文件夾列表、當(dāng)前路徑下文件名列表。

list的.append()、.extend()和.sort()方法都是原地修改，sorted()函數(shù)不是。

將list寫入.txt文件時(shí)需要把list轉(zhuǎn)為str，直接用str()函數(shù)強(qiáng)轉(zhuǎn)會(huì)很丑，用換行符連接list每個(gè)元素會(huì)好看很多。

os.path代表系統(tǒng)換行符，windows下為”\r\n”，其他系統(tǒng)多是”\n”。然而無論是用os.path還是”\n”連接列表元素，最后用windows記事本打開都一樣換行，但是用vs code打開的話os.path會(huì)多換一行即看上去一行間一行，這里面到底為啥有點(diǎn)說到，可能和Python的write機(jī)制有關(guān)，暫時(shí)不深究（留坑）。

關(guān)于文件讀寫，大多數(shù)資料推薦with as形式，確實(shí)更簡潔一點(diǎn)。

PS：

說os.walk(）暴力是因?yàn)樗娴陌凑漳夸洏浔闅v了所給路徑中的所有文件夾和文件，文件量大而所要找的文件名少的情況下會(huì)慢一些（其實(shí)我覺得慢不了多少），用os.listdir()寫成遞歸函數(shù)的話執(zhí)行效率可能會(huì)高一些，但是os.walk(）邏輯簡單好寫，各位隨意，我干了！

以上這篇Python3.遍歷某文件夾提取特定文件名的實(shí)例就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。

您可能感興趣的文章: