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