1、import os
2、OS模块下有两个函数:os.walk(),os.listdir()
root,dirs,files = os.walk(path)
a. os.walk() 返回值为当前path路径下的:目录路径、所有子目录、非目录子文件
扩展:获取目录下指定类型的文件
# -*- 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()函数将路径拆分为文件名+扩展名
b. os.listdir() 获取当前目录下的所有子文件(****不包含子目录下的文件****)
扩展:获取当前目录下所有文件
# -*- 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)