python读取文件名称生成list的方法

百变鹏仔 2小时前 #Python

文章标签文件名称

下面为大家分享一篇python读取文件名称生成list的方法，具有很好的参考价值，希望对大家有所帮助。一起过来看看吧

经常需要读取某个文件夹下所有的图像文件。

我使用python写了个简单的代码，读取某个文件夹下某个后缀的文件，将文件名生成为文本（csv格式）

import fnmatchimport osimport pandas as pdimport numpy as np import sysInputStra = 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

立即学习“Python免费学习笔记（深入）”；

使用时：

在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的方法

Python实现购物车程序

神经网络(BP)算法Python实现及应用

Numpy掩码式数组详解

对numpy中布尔型数组的处理方法

一份python入门应该看的学习资料