本文采用os.walk()和os.listdir()两种方法,获取指定文件夹下的文件名。
python os模块获取指定文件夹下所有文件名
第一种方法使用os.walk():
os.walk() 方法用于通过在目录树种游走输出在目录中的文件名,向上或者向下。
语法
walk()方法语法格式如下:os.walk(top[, topdown=True[, οnerrοr=None[, followlinks=False]]])
参数top -- 根目录下的每一个文件夹(包含它自己), 产生3-元组 (dirpath, dirnames, filenames)【文件夹路径, 文件夹名字, 文件名】。
topdown --可选,为True或者没有指定, 一个目录的的3-元组将比它的任何子文件夹的3-元组先产生 (目录自上而下)。如果topdown为 False, 一个目录的3-元组将比它的任何子文件夹的3-元组后产生 (目录自下而上)。
onerror -- 可选,是一个函数; 它调用时有一个参数, 一个OSError实例。报告这错误后,继续walk,或者抛出exception终止walk。
followlinks -- 设置为 true,则通过软链接访问目录。
样例代码: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):
li=[]
for root, dirs, files in os.walk(file_dir):
for file in files:
if os.path.splitext(file)[1] == '.jpeg':
li.append(os.path.join(root, file))
return li
其中os.path.splitext()函数将路径拆分为文件名+扩展名,例如os.path.splitext(“D:/.jpg”)将得到”D:/aaa“+".jpg"。
第二种方法使用os.listdir() 方法
os.listdir() 方法用于返回指定的文件夹包含的文件或文件夹的名字的列表。这个列表以字母顺序。 它不包括 '.' 和'..' 即使它在文件夹中。
语法
listdir()方法语法格式如下:os.listdir(path)
参数path -- 需要列出的目录路径
样例代码:import os, sys
# 打开文件
path = "/var/"
dirs = os.listdir( path )
# 输出所有文件和文件夹
for file in dirs:
print(file)