利用os.walk()遍历目录
import os
def get_file(folderpath, extendType=None):
'''
folderpath: 要查询的文件夹路径
extendType: 文件类型,如pdf
'''
count = 0
for fpath, dir_list, file_list in os.walk(folderpath):
for filename in file_list:
if extendType:
if filename.split('.')[-1]==extendType: #对文件名进行分割,精确匹配后缀,如1.docx,split会分割成列表['1','docx'],切片取出最后的后缀
count +=1
print(os.path.join(fpath, filename)) # 显示文件全路径和文件名称(包含扩展名)
else:
count += 1
print(os.path.join(fpath, filename))
if extendType:
print(f'{extendType}后缀文件共计:{count}个')
else:
print(f'该文件夹下共计文件:{count}个')
if __name__ == '__main__':
get_file('d:\\')
get_file('d:\\','pdf')
运行结果:
部分截图