python 如何输出文件名称

Python输出文件名称的方法有多种，例如使用os模块、glob模块、pathlib模块。 在本文中，我们将详细介绍这几种方法，并结合实际案例来展示如何使用它们。

一、使用os模块

os模块是Python的一个标准库模块，它提供了非常多的与操作系统交互的方法。使用os模块，我们可以很方便地获取目录中的文件名称。

1. 使用os.listdir()方法

os.listdir()方法用于返回指定路径下的文件和文件夹列表。我们可以遍历这个列表，并使用os.path.isfile()方法来判断是否为文件，从而获取文件名称。

import os
def list_files_in_directory(directory):
    try:
        files = os.listdir(directory)
        for file in files:
            if os.path.isfile(os.path.join(directory, file)):
                print(file)
    except FileNotFoundError:
        print(f"The directory {directory} does not exist")
directory = './example_directory'
list_files_in_directory(directory)

2. 使用os.walk()方法

os.walk()方法生成目录树下的所有文件名，它是一个生成器，返回一个三元组（dirpath, dirnames, filenames）。我们可以遍历这个生成器来获取所有文件的名称。

import os
def list_files_with_os_walk(directory):
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            print(os.path.join(dirpath, filename))
directory = './example_directory'
list_files_with_os_walk(directory)

二、使用glob模块

glob模块提供了一个函数用于查找符合特定规则的文件路径名。它支持通配符如*、?、[]等。

1. 使用glob.glob()方法

glob.glob()方法返回所有匹配的文件路径名。我们可以遍历这个列表来获取文件名称。

import glob
def list_files_with_glob(directory, pattern='*'):
    files = glob.glob(os.path.join(directory, pattern))
    for file in files:
        print(file)
directory = './example_directory'
list_files_with_glob(directory)

三、使用pathlib模块

pathlib模块是Python 3.4版本引入的一个面向对象的文件系统路径操作模块。它的语法更加直观和现代化。

1. 使用Path类

Path类提供了许多方法来操作文件和目录，我们可以使用它的iterdir()方法来遍历目录下的文件和文件夹，并使用is_file()方法来判断是否为文件。

from pathlib import Path
def list_files_with_pathlib(directory):
    path = Path(directory)
    for file in path.iterdir():
        if file.is_file():
            print(file.name)
directory = './example_directory'
list_files_with_pathlib(directory)

四、结合实际应用场景

1. 获取特定类型的文件

在实际应用中，我们可能需要获取特定类型的文件，例如所有的.txt文件。我们可以结合glob模块或pathlib模块来实现这个需求。

import glob
from pathlib import Path
def list_specific_type_files(directory, file_extension):
    # 使用glob模块
    files = glob.glob(os.path.join(directory, f'*.{file_extension}'))
    for file in files:
        print(file)
    # 使用pathlib模块
    path = Path(directory)
    for file in path.glob(f'*.{file_extension}'):
        print(file)
directory = './example_directory'
file_extension = 'txt'
list_specific_type_files(directory, file_extension)

2. 递归获取所有文件

有时候我们需要递归获取目录及其子目录下的所有文件，可以结合os.walk()方法或pathlib模块的rglob()方法来实现。

import os
from pathlib import Path
def list_all_files_recursively(directory):
    # 使用os.walk()方法
    for dirpath, dirnames, filenames in os.walk(directory):
        for filename in filenames:
            print(os.path.join(dirpath, filename))
    # 使用pathlib模块
    path = Path(directory)
    for file in path.rglob('*'):
        if file.is_file():
            print(file)
directory = './example_directory'
list_all_files_recursively(directory)

3. 异常处理

在实际开发中，处理异常是非常重要的，例如文件夹不存在或权限不足等。我们可以在代码中加入异常处理来提高程序的健壮性。

import os
from pathlib import Path
def list_files_with_exception_handling(directory):
    try:
        path = Path(directory)
        if not path.exists():
            raise FileNotFoundError(f"The directory {directory} does not exist")
        for file in path.iterdir():
            if file.is_file():
                print(file.name)
    except FileNotFoundError as fnf_error:
        print(fnf_error)
    except PermissionError as p_error:
        print(f"Permission denied: {p_error}")
directory = './example_directory'
list_files_with_exception_handling(directory)