码迷,mamicode.com
首页 > 其他好文 > 详细

os模块常见用法(一)

时间:2020-03-27 16:52:25      阅读:120      评论:0      收藏:0      [点我收藏+]

标签:port   path   lse   return   end   error   for   idc   file   

 

os.walk()

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

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)

# !/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(dirs) #当前路径下所有子目录 
print(files) #当前路径下所有非目录子文件

 

当需要特定类型的文件时

# !/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: 
            if os.path.splitext(file)[1] == .jpg: 
                FileList.append(os.path.join(root, file)) 
    return FileList

 


其中os.path.splitext()函数将路径拆分为文件名+扩展名,例如 os.path.splitext(“E:\IDCard\love.jpg”) 将得到 (E:\\IDCard\\love, .jpg) 

os.listdir()

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

直接给出代码,函数将返回类型为‘.jpg’个文件名:

# !/usr/bin/env python
# -*- 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]==.jpg: 
            list_name.append(file_path)

     

 

os模块常见用法(一)

标签:port   path   lse   return   end   error   for   idc   file   

原文地址:https://www.cnblogs.com/y-sh/p/12582204.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!