經常需要讀取某個文件夾下所有的圖像文件。
我使用python寫了個簡單的代碼,讀取某個文件夾下某個后綴的文件,將文件名生成為文本(csv格式)
import fnmatch import os import pandas as pd import numpy as np import sys InputStra = sys.argv[1] InputStrb = sys.argv[2] def ReadSaveAddr(Stra,Strb): #print(Stra) #print(Strb) print("Read :",Stra,Strb) a_list = fnmatch.filter(os.listdir(Stra),Strb) print("Find = ",len(a_list)) df = pd.DataFrame(np.arange(len(a_list)).reshape((len(a_list),1)),columns=['Addr']) df.Addr = a_list #print(df.head()) df.to_csv('Get.lst',columns=['Addr'],index=False,header=False) print("Write To Get.lst !") ReadSaveAddr(InputStra,InputStrb)
上面代碼保存為:GetLst.py
使用時:
在cmd窗口輸入:
python GetLst.py F:/train/pos *.png
發現上面代碼不能深入到下一層目錄,又做了點修改:
def ReadSaveAddr2(Stra,Strb): df = pd.DataFrame(np.arange(0).reshape(0,1),columns=['Addr']) print(df) path = InputStra for dirpath,dirnames,filenames in os.walk(path): #for filename in filenames: a_list = fnmatch.filter(os.listdir(dirpath),Strb) if len(a_list): dft = pd.DataFrame(np.arange(len(a_list)).reshape((len(a_list),1)),columns=['Addr']) dft.Addr = a_list dft.Addr = dirpath + '\\' + dft.Addr#輸出絕對路徑 frames = [df,dft] df = pd.concat(frames) print(df.shape) df.to_csv('Get.lst',columns='Addr'],index=False,header=False) print("Write To Get.lst !")
以上這篇python讀取文件名稱生成list的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持億速云。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。