python代码中可以通过下面的代码来获取指定文件或指定路径下的所有指定类型文件。

import glob
import os

# args.file 是指定的文件名称或者路径，如果是路径，会获取该路径下的指定类型的文件
fp_list = []
if os.path.isfile(args.file):
    fp_list.append(args.file)
elif os.path.isdir(args.file):
    fn_list = glob.glob1(args.file, '*g')    # 获取以g结尾的文件名称
    fp_list = [os.path.join(args.file, fn) for fn in fn_list]

glob使用方法

glob.glob()

#获取指定目录下的所有图片
print (glob.glob(r"/home/qiaoyunhao/*/*.png"),"\n")#加上r让字符串不转义

glob.iglob()

#父目录中的.py文件
f = glob.iglob(r'../*.py')
print (f) #<generator object iglob at 0x00B9FF80>
for py in f:
    print (py)

glob.glob1()

fn_list = glob.glob1(args.file, '*g')    # 获取以g结尾的文件名称