遇一山,过一山,处处有风景;只要勇敢向前,一路尽是繁花盛开。 | (点击查看→)【测试干货】python/java自动化、持续集成、性能、测开、简历、笔试面试等

python获取当前文件夹下所有文件名

os 模块下有两个函数:

os.walk()

os.listdir()

 

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os

def file_name(file_dir):
    for root, dirs, files in os.walk(file_dir):
        print(root) #当前目录路径(包含所有子目录)
        # print("===============")
        # print(dirs) #当前路径下所有子目录(同一路径下的存一个列表中)
        # print("===============")
        # print(files) #当前路径下所有非目录子文件(同一路径下的存一个列表中)
file_name("E:\\qzcsbj")

  

获取路径下指定格式的文件

方式一:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os

def file_name(file_dir):
    fileList = []
    for root, dirs, files in os.walk(file_dir):
        for file in files:
            # print(file)
            if os.path.splitext(file)[1] == '.txt':  # os.path.splitext()函数将路径拆分为文件名+扩展名
            # if file.split('.')[1].strip() == 'txt':
                fileList.append(os.path.join(root, file))
    return fileList


res = file_name("E:\\qzcsbj")
print(res)

  

方式二:

#!/usr/bin/env python
# -*- coding: utf-8 -*-

import os

def list_dir(path, list_name):  # 传入存储的list
	for file in os.listdir(path):  # os.listdir(path),路径下的文件及文件夹,不包含子文件和子文件夹
		file_path = os.path.join(path, file)
		if os.path.isdir(file_path):  # 判断是否目录
			list_dir(file_path, list_name)
		else:
			list_name.append(file_path)


fileList=[]
list_dir("E:\\qzcsbj", fileList)
print(fileList)

 

 整理自网络,如有侵权,请联系删除。
 
 
posted @ 2017-12-17 15:56  全栈测试笔记  阅读(15285)  评论(0编辑  收藏  举报
浏览器标题切换
浏览器标题切换end