Python os.path.listdir使用

小蓝 • 2024-12-31 11:49 • 编程

介绍

Python语言中，使用os模块可以对操作系统进行访问，包括文件、文件夹等访问。os.path模块提供了一些函数，用于处理路径字符串。os.path.listdir()函数用于获取指定路径下的所有文件和文件夹的名称，并以列表的形式返回。

正文

一、listdir()函数的使用方法

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

import os
os.listdir(path)

其中，path是需要获取文件名和文件夹名的路径。如果想要获取当前路径下的文件和文件夹，可以使用os.getcwd()函数获取路径。

函数返回的结果是一个列表，包含了指定路径下所有文件名和文件夹名。下面是一个例子：

import os
current_folder_path = os.getcwd()
files_and_folders = os.listdir(current_folder_path)
print(files_and_folders)

如果将代码保存该文件夹下，并执行该代码，结果可能类似于：

['file1.jpg', 'file2.txt', 'folder1', 'folder2']

二、listdir()函数的应用场景

os.path.listdir()函数可以用于很多场景，比如：

1. 获取指定文件夹下的所有文件名和文件夹名

可以使用os.path.listdir()函数获取指定文件夹下的所有文件名和文件夹名，并对这些元素进行处理。

import os
folder_path = 'folder1'
files_and_folders = os.listdir(folder_path)
print(files_and_folders)

结果可能如下：

['file1.jpg', 'file2.txt']

2. 遍历文件夹的所有文件名和文件夹名

可以使用os.path.foreach()函数获取指定文件夹下的所有文件名和文件夹名，并利用递归函数对文件夹进行遍历。

import os

def traverse_folder(folder_path):
    files_and_folders = os.listdir(folder_path)
    for file_or_folder_name in files_and_folders:
        file_or_folder_path = os.path.join(folder_path, file_or_folder_name)
        if os.path.isdir(file_or_folder_path):
            traverse_folder(file_or_folder_path)
        else:
            print(file_or_folder_path)

current_folder_path = os.getcwd()
traverse_folder(current_folder_path)

3. 根据文件名和文件夹名进行筛选

可以使用os.path.listdir()函数获取指定文件夹下的所有文件名和文件夹名，并利用if语句筛选需要的文件名和文件夹名。

import os
folder_path = 'folder1'
files_and_folders = os.listdir(folder_path)
for file_or_folder_name in files_and_folders:
    if file_or_folder_name.endswith('.txt'):
        print(file_or_folder_name)

结果可能如下：

['file2.txt']

小结

os.path.listdir()函数是Python中获取文件名和文件夹名的重要函数，可以用于很多场景。我们可以结合其他函数和语句，对文件和文件夹进行遍历、筛选等操作。在进行路径操作时，建议使用os.path模块提供的函数，因为这些函数会考虑跨平台的问题，能够保证在不同的操作系统上都能正常工作。

原创文章，作者：小蓝，如若转载，请注明出处：https://www.506064.com/n/303239.html