os 模块下有两个函数:
os.walk()
os.listdir()
1 # -*- coding: utf-8 -*-
2
3 import os
4
5 def file_name(file_dir):
6 for root, dirs, files in os.walk(file_dir):
7 print(root) #当前目录路径
8 print(dirs) #当前路径下所有子目录
9 print(files) #当前路径下所有非目录子文件
# 只获取当前路径下文件名,不获取文件夹中文件名
1 # -*- coding: utf-8 -*-
2
3 import os
4
5 def file_name(file_dir):
6 L=[]
7 for root, dirs, files in os.walk(file_dir):
8 for file in files:
9 if os.path.splitext(file)[1] == '.jpeg': # 想要保存的文件格式
10 L.append(os.path.join(root, file))
11 return L
12
13
14 #其中os.path.splitext()函数将路径拆分为文件名+扩展名
# 递归获取路径下所有文件名
1 # -*- coding: utf-8 -*-
2 import os
3
4 def listdir(path, list_name): #传入存储的list
5 for file in os.listdir(path):
6 file_path = os.path.join(path, file)
7 if os.path.isdir(file_path):
8 listdir(file_path, list_name)
9 else:
10 list_name.append(file_path)
网友评论