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列表insert()函數(shù)的使用方法，文章通過代碼示例介紹的非常詳細(xì)，對我們學(xué)習(xí)或工作有一定的參考價值,需要的朋友可以參考下2023-07-07
這篇文章主要介紹了Python實現(xiàn)讀取文件最后n行的方法,涉及Python針對文件的讀取、遍歷與運算相關(guān)操作技巧,需要的朋友可以參考下2017-02-02
Python?time時間格式化和設(shè)置時區(qū)實現(xiàn)代碼詳解
這篇文章主要介紹了Python?time時間格式化和設(shè)置時區(qū)實現(xiàn)代碼，文中通過示例代碼介紹的非常詳細(xì)，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值2023-02-02
今天小編就為大家分享一篇使用python的pyplot繪制函數(shù)實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02
利用Python繪制Jazz網(wǎng)絡(luò)圖的例子
今天小編就為大家分享一篇利用Python繪制Jazz網(wǎng)絡(luò)圖的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧2019-11-11
這篇文章主要介紹了python實現(xiàn)AdaBoost算法的示例，幫助大家更好的理解和了解機器學(xué)習(xí)算法，感興趣的朋友可以了解下2020-10-10
python?中defaultdict()對字典進行初始化的用法介紹
這篇文章主要介紹了python?中defaultdict()對字典進行初始化,一般情況下，在使用字典時，先定義一個空字典（如dict_a?=?{}），然后往字典中添加元素只需要?dict_a[key]?=?value即可，本文通過實例代碼介紹具體用法，需要的朋友可以參考下2022-07-07
私有化是為了實現(xiàn)更好的封裝形式。能隱藏屬性，不被隨意修改。以前置雙下劃線__屬性/方法名定義，但是結(jié)尾不能是雙下劃線。如果不讓某些屬性在類外部被隨意更改，可設(shè)置成私有屬性。在類定義的內(nèi)部可以訪問2023-02-02
Cookie用于服務(wù)器實現(xiàn)會話，用戶登錄及相關(guān)功能時進行狀態(tài)管理,這篇文章主要介紹了使用python處理cookie的方法,感興趣的小伙伴們可以參考一下2016-02-02
這篇文章主要介紹了對于Python異常處理方法的建議,摘選自StackOverflow上的熱門問題的回答,闡述了except:pass的使用時需要注意的地方,需要的朋友可以參考下2015-04-04