方案 1:利用 glob 模块
import glob
import os
dst_path = 'C:/'
ext_name = '*.txt'
os.chdir( dst_path )
for file in glob.glob( ext_name ):
pass
方案 2:利用 os.listdir
import os
dst_path = 'd:/'
ext_name = 'txt'
for file in os.listdir( dst_path ):
if file.endswith( ext_name ):
print( file )
方案 3: 利用 os.walk 遍历指定路径下所有文件(含子路径,与方案 1 & 2 有显著差异)
import os
dst_path = 'd:/'
ext_name = 'txt'
for r, d, files in os.walk( dst_path ):
for file in files:
if file.endswith( ext_name ):
print( file )