python 寻找文件夹下含有相同字符的所有文件名

最近在做一个项目时,需要把某文件夹下的含有某些相同字符的文件的文件名字读取并生成一个列表文件:如图寻找下面文件夹中全部的含“traindata_017”的所有文件名字。

《python 寻找文件夹下含有相同字符的所有文件名》

代码:

def readTfrecordNames(halfFileName):
    listname = []
    for filename in glob.glob('./*'):
    #print(filename)
        if halfFileName in filename:
            filename = filename.split('\\')[-1] # .\traindata_017.tfrecords 去除前面的 .\
            #print(filename)
            listname.append(filename)
    return listname

listnames = readTfrecordNames('traindata_017')

 

点赞