本文采用os.walk()和os.listdir()两种方法,获取指定文件夹下的文件名。
python os模块获取指定文件夹下所有文件名
第一种方法使用os.walk():
os.walk() 方法用于通过在目录树种游走输出在目录中的文件名,向上或者向下。
语法
walk()方法语法格式如下:
os.walk(top[, topdown=True[, onerror=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)
第三种使用列表推导式-获取每个文件的具体路径
import os root="/home/chenxinming" file_names=[os.path.join(path, name) for path, subdirs, files in os.walk(root) for name in files]
第四种只获取文件名
from os import listdir from os.path import isfile, join root="/home/chenxinming" filenames = [f for f in listdir(root) if isfile(join(root, f))] print(filenames [:10]) >>>['.profile', '.gitignore', '.mysql_history', '.bash_history',]
第五种更精简版本只获取文件名(只支持python3.x)
from os import walk root="/home/chenxinming" _, _, filenames = next(walk(root)) print(filenames [:10]) >>>['.profile', '.gitignore', '.mysql_history', '.bash_history',]
还没有留言,还不快点抢沙发?