os 模块下有两个函数:
os.walk()
这个可以分类获取该目录下的各种文件和文件夹。
这个可以用来对文件名或类别进行判断,以获取需要的文件。
这个可以递归获取根目录下所有的文件包括子文件夹中的文件。
os.walk()
os.listdir()
- # -*- coding: utf-8 -*-
- import os
- def file_name(file_dir):
- for root, dirs, files in os.walk(file_dir):
- print(root) #当前目录路径
- print(dirs) #当前路径下所有子目录
- print(files) #当前路径下所有非目录子文件
- # -*- coding: utf-8 -*-
- import os
- def file_name(file_dir):
- L=[]
- for root, dirs, files in os.walk(file_dir):
- for file in files:
- if os.path.splitext(file)[1] == '.jpeg':
- L.append(os.path.join(root, file))
- return L
- #其中os.path.splitext()函数将路径拆分为文件名+扩展名
- # -*- coding: utf-8 -*-
- import os
- def listdir(path, list_name): #传入存储的list
- for file in os.listdir(path):
- file_path = os.path.join(path, file)
- if os.path.isdir(file_path):
- listdir(file_path, list_name)
- else:
- list_name.append(file_path)