python获取指定目录下特定格式的文件名

之前一直用windows下的bat脚本获取一个目录下的指定格式的文件名，如下所示：

dir *.jpg /b/s > train.set

pause

十分简单，将这个bat文件放到你想要获取文件名的目录下，然后双击运行就可以将你想要保存的文件名写在train.set文件之中了。

今天由于实际需求，这个功能需要用python来实现，在网上查了一下发现，python脚本实现起来也很方便，而且也比较灵活，如下所示：

# -*- coding: utf-8 -*-
"""
Created on Mon Dec 12 14:59:46 2016

@author: shenruixue

to calculate size after filter in conv and pool
"""
import os.path, time
import exceptions
import glob
import os


class TypeError (Exception):
  pass
if __name__ == '__main__':
 file_srx = open("train.set")#其中包含所有待计算的文件名
 file_object = open('all_time.txt', 'w')
 line = file_srx.readline()
 diff_time_all = 0
 while line:
  f = line[:-1]    # 除去末尾的换行符
  print (f)
  print ('***********************************************************')
  mtime = time.ctime(os.path.getmtime(f))
  ctime = time.ctime(os.path.getctime(f))
  mtime_s = (os.path.getmtime(f))
  ctime_s = (os.path.getctime(f))
  print "Last modified : %s, last created time: %s" % (mtime, ctime)
  diff_time = (int(mtime_s) - int(ctime_s))
  diff_time_all = diff_time_all + diff_time
  print "diff time is ", diff_time
  
  f_record = f[:-17] #根据f给f_record进行赋值，需要根据实际情况修改17的大小
  print (f_record)
  print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*')
  os.chdir(f_record) 
  for file_name in glob.glob("*.jpg"):
   print file_name
   print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*')
  
  file_object.write(file_name + " time is " + str(diff_time) + "
")
  line = file_srx.readline() 
 print "diff_time_all is ", diff_time_all
 
 file_object.write(str(diff_time_all))
 file_object.close( )

可以看到用到了os与glog两个包，很方便快捷。

可以写成一个小脚本，以后方便使用。如下：

# -*- coding: utf-8 -*-
"""
Created on Mon Dec 12 14:59:46 2016

@author: shenruixue

to calculate size after filter in conv and pool
"""
import os.path, time
import exceptions
import glob
import os
rootdir = os.getcwd()
if __name__ == '__main__':
  file_object = open('name.set', 'w')
  for file_name in glob.glob("*.jpg"):
   print file_name
   print ('*~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~*')
   file_object.write(rootdir + file_name + "
")
  file_object.close( )