您的位置:

Python中os.scandir的使用方法

介绍

Python是一种面向对象、解释型、动态类型的高级程序设计语言,其包含许多强大的模块和库,其中之一就是os.scandir模块。os.scandir()函数是Python 3.5版本引入的,用于返回指定路径下的文件和目录信息,包括目录项的名称、类型、大小等信息。使用os.scandir()函数可以在遍历文件夹时更快更稳定,因为它可以同时获取目录项的元数据和文件名。

正文

一、扫描目录

使用os.scandir()函数扫描目录很简单,只需要将目录作为参数传递给os.scandir()函数即可。以下是扫描包含文件和目录的目录的示例代码:

import os

path = '/path/to/directory/'
with os.scandir(path) as entries:
    for entry in entries:
        print(entry.name)

在上面的代码中,我们首先指定了要扫描的目录的路径。然后,使用with语句打开一个目录句柄,并在该目录中遍历每个目录项,依次处理每个目录项的name属性,打印出所有的文件和目录名称。

二、筛选扫描结果

os.scandir()函数返回一个目录迭代器,可以对迭代器进行筛选,只选择需要的文件或文件夹。以下是只列出目录中的文件的示例代码:

import os

path = '/path/to/directory/'
with os.scandir(path) as entries:
    for entry in entries:
        if entry.is_file():
            print(entry.name)

在上面的代码中,我们使用is_file()方法来筛选所有文件,而不是目录。

三、获取文件属性

os.scandir()函数还返回文件的多个属性,例如文件大小、修改时间、创建时间等。以下是获取文件属性的示例代码:

import os

path = '/path/to/file'
with os.scandir(path) as entry:
    if entry.is_file():
        print(f"名称:{entry.name}")
        print(f"路径:{entry.path}")
        print(f"大小:{entry.stat().st_size}")
        print(f"修改时间:{entry.stat().st_mtime}")
        print(f"创建时间:{entry.stat().st_ctime}")

在上面的代码中,我们首先使用is_file()方法检查目录项是否为文件,然后使用entry.name、entry.path、entry.stat().st_size和entry.stat().st_mtime等属性获取文件的名称、路径、大小、修改时间和创建时间。

四、递归遍历目录

os.scandir()函数和os.listdir()函数一样,只返回指定目录下的文件和目录信息。如果要遍历目录树以及包含的所有子目录,可以使用os.walk()函数。以下是使用os.scandir()和os.walk()函数递归遍历目录的示例代码:

import os

def scan_dir(dir_path):
    with os.scandir(dir_path) as entries:
        for entry in entries:
            if entry.is_dir():
                print(f"文件夹: {entry.name}")
                scan_dir(entry.path)
            else:
                print(f"文件: {entry.name}")

root_dir = '/path/to/root/dir'
scan_dir(root_dir)

在上面的代码中,我们使用scan_dir()函数来递归地遍历所有子目录。如果目录项是文件夹,进行处理,否则打印出该目录项的名称。

总结

os.scandir()函数是Python 3.5版本引入的,是一种快速、可靠的方法,用于获取指定路径下的文件和目录信息。与os.listdir()函数相比,os.scandir()函数可以返回目录项的元数据和文件名,使文件和目录迭代更快更稳定。