os模块常见用法

Posted 2021-03-03 y-sh

 

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:IDCardlove.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)