Python批量获取指定后缀文件名并导出CSV的方法

星野

发布时间：2026-10-13 | 3人浏览过

原创

本文介绍如何使用Python脚本批量读取指定文件夹下的图像文件，并根据后缀名过滤生成文件名列表。通过fnmatch和os模块实现基础功能，并优化支持递归遍历子目录，最终将结果保存为CSV格式，便于后续处理。

在实际开发中，经常需要获取某个文件夹内所有的图像文件。

这里提供一个简单的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读取文件名生成列表的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持本站。