我想创建一个函数，在输入中输入文件名，作为输出，我将获取列表中的所有图像。在这个文件中，我有各种文档（文本、图像、视频……），并且所有图像都有不同的扩展名。我写了这段代码，但它似乎不起作用
def readFileImages(strFolderName):
print strFolderName
image_list = []
st=strFolderName+"*.png"
for filename in glob.glob(st): #assuming gif
image_list.append(filename)
return image_list
您需要确保您有一个路径分隔符。最好使用os.path.join()为您添加它来完成：
import glob
import os
def readFileImages(strFolderName):
print strFolderName
image_list = []
st = os.path.join(strFolderName, "*.gif")
for filename in glob.glob(st): #assuming gif
image_list.append(filename)
return image_list
print readFileImages(os.getcwd()) # Use current working directory as example
请注意，glob.glob()已经返回一个列表，因此以下内容是等效的：
import glob
import os
def readFileImages(strFolderName):
st = os.path.join(strFolderName, "*.gif")
return glob.glob(st)
print readFileImages(os.getcwd())
os.path.join()更好的原因是如果strFolderName不包含拖尾分隔符，它将被添加。如果确实有一个，则不会添加额外的一个。此外，如果您在不同的操作员系统上运行脚本，将自动使用正确的分隔符。你可以看到哪个将被使用print os.path.sep
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句