Python获取指定文件夹下的文件名的方法_Python

本文采用os.walk()和os.listdir()两种方法，获取指定文件夹下的文件名。

一、os.walk()

模块os中的walk()函数可以遍历文件夹下所有的文件。

1	`os.walk(top, topdown=Ture, onerror=None, followlinks=False)`

该函数可以得到一个三元tupple(dirpath, dirnames, filenames).

参数含义：

dirpath：string，代表目录的路径；
dirnames：list，包含了当前dirpath路径下所有的子目录名字（不包含目录路径）；
filenames：list，包含了当前dirpath路径下所有的非目录子文件的名字（不包含目录路径）。

注意，dirnames和filenames均不包含路径信息，如需完整路径，可使用os.path.join(dirpath, dirnames)

下面给出代码;

									# -*- 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()函数将路径拆分为文件名+扩展名，例如os.path.splitext(“E:/lena.jpg”)将得到”E:/lena“+".jpg"。

二、os.listdir()

os.listdir()函数得到的是仅当前路径下的文件名，不包括子目录中的文件，所有需要使用递归的方法得到全部文件名。

直接给出代码，函数将返回类型为‘.jpeg'个文件名：

									# -*- coding: utf-8 -*- 

									import os 

									def listdir(path, list_name): 

									  for file in os.listdir(path): 

									    file_path = os.path.join(path, file) 

									    if os.path.isdir(file_path): 

									      listdir(file_path, list_name) 

									    elif os.path.splitext(file_path)[1]=='.jpeg': 

									      list_name.append(file_path)

初学python，犯了些低级错误，浪费了很多时间，首先一定要确保代码的四空格缩进，注意直接回车下一行并不是四空格缩进；还有一点是在命令提示符窗口（Windows）使用python解释器第一次运行代码后，修改源代码，如若再次运行需要退出重新进入python解释器，否则仍将运行第一次执行过的代码。

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持服务器之家。

原文链接：http://blog.csdn.net/lsq2902101015/article/details/51305825

Python获取指定文件夹下的文件名的方法

相关文章

热门资讯