python讀取文件名稱生成list的方法
更新時間：2018年04月27日 10:24:08 作者：masbbx123
下面小編就為大家分享一篇python讀取文件名稱生成list的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
經(jīng)常需要讀取某個文件夾下所有的圖像文件。
我使用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
發(fā)現(xiàn)上面代碼不能深入到下一層目錄，又做了點修改：
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的方法就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
python中的單引號雙引號區(qū)別知識點總結(jié)
在本篇文章中小編給大家整理了關(guān)于python中的單引號雙引號有什么區(qū)別的相關(guān)知識點以及實例代碼，需要的朋友們參考下。2019-06-06
這篇文章主要給大家介紹了關(guān)于python實現(xiàn)棋盤覆蓋問題及可視化的相關(guān)資料，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧2021-03-03
這篇文章主要分享的是Python常見的幾種數(shù)據(jù)加密方式，主要包括線性散列算法（簽名算法）MD5，sha1、對稱性加密算法?AES?DES、非對稱性加密算法?RSA，具體詳細內(nèi)容介紹，需要的小伙伴可以參考一下2022-06-06
今天小編就為大家分享一篇Python測試Kafka集群(pykafka)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-12-12
如何利用Python+OpenCV實現(xiàn)簡易圖像邊緣輪廓檢測(零基礎(chǔ))
輪廓是形狀分析和物體檢測和識別的有用工具,下面這篇文章主要給大家介紹了關(guān)于如何利用Python+OpenCV實現(xiàn)簡易圖像邊緣輪廓檢測(零基礎(chǔ))的相關(guān)資料,文中通過實例代碼介紹的非常詳細,需要的朋友可以參考下2022-05-05
今天小編就為大家分享一篇pycharm new project變成灰色的解決方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-06-06
WINDOWS 同時安裝 python2 python3 后 pip 錯誤的解決方法
這篇文章主要給大家分享的是在WINDOWS下同時安裝 python2 python3 后 pip 錯誤的解決方法，非常的實用，有需要的小伙伴可以參考下2017-03-03
這篇文章主要為大家詳細介紹了樹莓派采用socket方式文件傳輸，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-06-06
wxpy基于itchat，使用了 Web 微信的通訊協(xié)議，通過大量接口優(yōu)化提升了模塊的易用性，并進行豐富的功能擴展。這篇文章主要介紹了python實現(xiàn)微信自動回復機器人功能,需要的朋友可以參考下2019-07-07
python中pandas對多列進行分組統(tǒng)計的實現(xiàn)
分組統(tǒng)計在很多時候都需要用到，可以實現(xiàn)很多數(shù)據(jù)庫函數(shù)的功能。本文主要介紹了python中pandas對多列進行分組統(tǒng)計的實現(xiàn)，感興趣的可以了解一下2021-06-06