blueice 发表于 2015-12-2 13:59:32

使用Python脚本获取指定格式文件列表的方法

在Python环境下获取指定后缀文件列表的方式。
  来源stackoverflow
  这里简单以*.txt的作为例子。


[*]使用glob(推荐)


1 import glob, os
2 os.chdir("/mydir")
3 for file in glob.glob("*.txt"):
4   print(file)
[*]简单实用os.listdir


1 import os
2 for file in os.listdir("./mydir"):
3   if file.endswith(".txt"):
4         print(file)
5 #或者使用函数式编程语句
6import os
7path = './mydir'
8text_files =
  

[*]如果需要遍历目录,使用os.walk


1import os
2for root, dirs, files in os.walk("/mydir"):
3      for file in files:
4          if file.endswith(".txt"):
5               print(os.path.join(root, file))
  
  http://www.xiami.com/widget/0_3480569/singlePlayer.swf

页: [1]
查看完整版本: 使用Python脚本获取指定格式文件列表的方法