一文深入学习Python中的os.listdir函数

1. 引言

在Python中，文件和目录操作是常见的任务之一。而os.listdir()函数是Python中用于获取指定目录下所有文件和子目录的函数之一。本篇博客将深入探索os.listdir()函数的用法和注意事项。

2. os模块简介

Python的os模块是与操作系统交互的一个接口，提供了许多与文件和目录操作相关的函数。通过os模块，我们可以执行文件和目录的创建、删除、重命名等操作。os.listdir()函数是os模块中的一个重要函数，用于获取指定目录下的所有文件和子目录。

3. os.listdir函数的基本用法

os.listdir()函数的基本语法如下：

import os
file_list = os.listdir(path)

其中，path为要获取文件和子目录列表的目录路径。如果不提供path参数，则默认为当前工作目录。该函数会返回一个包含指定目录下所有文件和子目录名称的列表。

下面是一个示例代码，演示了如何使用os.listdir()函数获取当前目录下的所有文件和子目录：

import os

file_list = os.listdir()

for file in file_list:
    print(file)

4. 获取指定目录下的文件和子目录

如果我们想要获取指定目录下的文件和子目录，可以将目录路径作为path参数传递给os.listdir()函数。

下面是一个示例代码，演示了如何使用os.listdir()函数获取指定目录下的所有文件和子目录：

import os
path = '/path/to/directory'
file_list = os.listdir(path)
for file in file_list:
    print(file)

5. 遍历目录树

有时候，我们需要遍历整个目录树，即获取指定目录及其子目录下的所有文件和子目录。可以使用递归的方法来实现这个功能。

下面是一个示例代码，演示了如何使用os.listdir()函数递归遍历目录树：

import os
def list_files(path):
    file_list = os.listdir(path)
    for file in file_list:
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            list_files(file_path)
        else:
            print(file_path)
path = '/path/to/directory'
list_files(path)

6. 处理隐藏文件和特殊文件

有些情况下，我们可能需要处理隐藏文件和特殊文件，例如符号链接等。可以使用os.listdir()函数结合其他函数来实现对隐藏文件和特殊文件的处理。

下面是一个示例代码，演示了如何使用os.listdir()函数过滤隐藏文件和特殊文件：

import os
def list_files(path):
    file_list = os.listdir(path)
    for file in file_list:
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            list_files(file_path)
        elif not file.startswith('.'):  # 过滤隐藏文件
            print(file_path)
path = '/path/to/directory'
list_files(path)

7. 错误处理和异常情况

在使用os.listdir()函数时，可能会遇到一些错误和异常情况。例如，如果指定的路径不存在或不是一个目录，os.listdir()函数将会抛出一个FileNotFoundError异常。

为了处理这些错误和异常，我们可以使用try-except语句来捕获并处理异常。

下面是一个示例代码，演示了如何处理FileNotFoundError异常：

import os
def list_files(path):
    try:
        file_list = os.listdir(path)
        for file in file_list:
            file_path = os.path.join(path, file)
            if os.path.isdir(file_path):
                list_files(file_path)
            else:
                print(file_path)
    except FileNotFoundError:
        print(f"Directory '{path}' does not exist.")
path = '/path/to/nonexistent_directory'
list_files(path)

8. os.scandir函数的替代选择

在Python 3.5及以上版本中，引入了os.scandir()函数，该函数相比于os.listdir()函数具有一些优势。os.scandir()函数返回一个迭代器，可以更高效地遍历目录并获取文件和子目录的详细信息。

下面是一个示例代码，演示了如何使用os.scandir()函数获取指定目录下的所有文件和子目录：

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

9. 总结

在本篇博客中，我们深入探索了Python中的os.listdir()函数。我们了解了该函数的基本用法和参数，以及如何获取指定目录下的文件和子目录，遍历目录树，处理隐藏文件和特殊文件，以及错误处理和异常情况。此外，我们还介绍了Python 3.5引入的os.scandir()函数，作为os.listdir()的替代选择。

os.listdir()函数在文件和目录操作中非常有用，可以帮助我们快速获取指定目录下的所有文件和子目录。然而，在使用该函数时，需要注意错误处理和异常情况，并根据实际需求选择合适的函数。

10. 参考文献

• Python官方文档：os.listdir()
• Python官方文档：os.scandir()

附：排序问题

当我们用os.listdir()打开文件时：

 发现返回的是这样的

['1.jpg', '10.jpg', '11.jpg', '12.jpg', '2.jpg', '20.jpg', '21.jpg', '22.jpg']

是按照采取了按字符键值排序的手段，所以10.jpg排在了前面。

解决办法：

返回的值都是数列了

可以按照自己的想法来进行“sort”排序

1.数字大小排序

按照“.”之前的数字大小排序

import os
path = r"test"
filenames=os.listdir(path)
filenames.sort(key=lambda x:int(x.split('.')[0]))
print(filenames)

得到

['1.jpg', '2.jpg', '10.jpg', '11.jpg', '12.jpg', '20.jpg', '21.jpg', '22.jpg']

2.在指定长度数字

某些文件包含字母或汉字

指定某几个位置的数字经行“sort”

import os
path = r"test"
filenames=os.listdir(path)
filenames.sort(key=lambda x:int(x[5:-4]))
print(filenames)

['屏幕截图 1.png', '屏幕截图 2.png', '屏幕截图 10.png', '屏幕截图 11.png', '屏幕截图 19.png', '屏幕截图 20.png']

要看看你的非数字字符的位置，来判断取值范围

到此这篇关于Python中os.listdir函数的文章就介绍到这了,更多相关Python os.listdir函数内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

最新评论