#找出指定目录下指定后缀的文件
def find_file(path,file_suffix):
#输入路径及要查找的文件后缀,file_suffix表示文件后缀
files = []
for file in os.listdir(path):
if file_suffix in os.path.splitext(file)[1]:
files.append(path+' '+file)
else:
return '没有此类型文件'
return files
print(find_file('D:/songhuajie/efforts/Python/test','txt'))
print(find_file('D:/songhuajie/efforts/Python/test','py'))