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結(jié)尾的文件名稱
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讓字符串不轉(zhuǎn)義
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結(jié)尾的文件名稱
