python读取文件名称生成list

最新推荐文章于 2023-11-28 21:29:29 发布

masbbx123

最新推荐文章于 2023-11-28 21:29:29 发布

阅读量5.8k

点赞数 2

CC 4.0 BY-SA版权

分类专栏： python

本文链接：https://blog.youkuaiyun.com/masbbx123/article/details/78954294

python 专栏收录该内容

20 篇文章

订阅专栏

本文介绍了一种使用Python批量读取指定目录下所有图像文件的方法，并将文件名以CSV格式保存。通过递归遍历子目录，确保了所有符合条件的图像文件都被包含。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

经常需要读取某个文件夹下所有的图像文件。
我使用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 !")