下面为大家分享一篇python读取文件名称生成list的方法，具有很好的参考价值，希望对大家有所帮助。一起过来看看吧
经常需要读取某个文件夹下所有的图像文件。
我使用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 读取指定文件夹下的所有图像方法
以上就是python读取文件名称生成list的方法的详细内容。