最近在做一個項目時,需要把某文件夾下的含有某些相同字符的文件的文件名字讀取並生成一個列表文件:如圖尋找下面文件夾中全部的含“traindata_017”的所有文件名字。
代碼:
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')