列出文件夹下所有的文本文件名python

Python033

列出文件夹下所有的文本文件名python,第1张

src_dir = r'./' # 源文件目录地址

def list_all_files(rootdir):

import os

_files = []

#列出文件夹下所有的目录与文件

list_file = os.listdir(rootdir)

for i in range(0,len(list_file)):

# 构造路径

path = os.path.join(rootdir,list_file[i])

# 判断路径是否是一个文件目录或者文件

# 如果是文件目录，继续递归

if os.path.isdir(path):

_files.extend(list_all_files(path))

if os.path.isfile(path):

_files.append(path)

return _files

files = list_all_files(src_dir)

print(files)

import os

os.chdir("D:\\python") #设置需要查看的目录

a = [x for x in os.listdir() if os.path.isdir(x)]

列表a里面就是当前目录下的文件夹名字

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

一、os.walk()

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

[python] view plain copy

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)

下面给出代码

[python] view plain copy

# -*- 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) #当前路径下所有非目录子文件

当需要特定类型的文件时，代码如下：

[python] view plain copy

# -*- 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’个文件名：

[python] view plain copy

# -*- 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)