import os
file_dir = 'D:/images/'
files_list = os.listdir(file_dir)
files=[]
n=0
if len(files_list) > 0:
print("当前文件夹不为空!!!")
print(files_list)
for k in files_list:
print(k)
# 提取文件夹内所有文件的后缀
print(os.path.splitext(k)[1])
files = os.path.splitext(k)[1]
if files == ".jpg":
n += 1
print(n)
if n >= 1:
print("文件夹中存在%i张“jpg”格式图片" % n)
else:
print("当前文件夹为空!!!")
os.path.splitext() 函数主要是分离文件的文件名和文件后缀,例如:
filename="1.jpg"
print(os.path.splitext(filename))
输出为:
('1', '.jpg')