def picture_name(path, list_name): #获取图片的名称
for file in os.listdir(path):
# file_path = os.path.join(path, file)
file_path = file
if os.path.isdir(file_path):
picture_name(file_path, list_name)
elif os.path.splitext(file_path)[1]=='.PNG':
list_name.append(file_path)
return list_name
if __name__ == "__main__":
# print(os.path.dirname(os.path.abspath(__file__)))
# file_name()
a = os.path.dirname(os.path.abspath(__file__))
c = listdir(a,[])
# print(c)
for path in c:
print(path)
用到os.listdir()函数得到的是仅当前路径下的文件名,不包括子目录中的文件,需要使用递归的方法得到全部文件名。
返回.PNG图片的名称。