python读取文件名称生成list

经常需要读取某个文件夹下所有的图像文件。
使用python写了个简单的代码，读取某个文件夹下某个后缀的文件，将文件名生成为文本（csv格式）

import fnmatch
import os
import pandas as pd
import numpy as np  
import sys

InputStra = 'LoveDA/Train/Urban/masks_png'  #相对路径
InputStrb = '*.png'

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)

上面代码保存为：image_mask.py
点击run