python用路径如何获取文件名

通过Python路径获取文件名的方式有多种，可以使用os.path模块、pathlib模块、os.scandir()方法等，其中os.path.basename()方法最为常用。 在本文中，我们将详细介绍这些方法，并通过示例代码帮助你更好地理解如何在Python中从路径获取文件名。

一、OS.PATH 模块

os.path 模块是Python标准库的一部分，提供了许多有用的函数来操作路径名。获取文件名的最简单方法是使用 os.path.basename() 函数。

import os
path = "/home/user/documents/report.pdf"
filename = os.path.basename(path)
print(filename)  # 输出：report.pdf

1、os.path.basename() 方法

os.path.basename(path) 方法会返回路径中的文件名部分。它会自动处理路径分隔符，无论是在Windows还是Unix系统中都适用。

import os
def get_filename(path):
    return os.path.basename(path)
print(get_filename("/home/user/documents/report.pdf"))  # 输出：report.pdf

2、os.path.splitext() 方法

os.path.splitext(path) 方法会将文件名和扩展名分开，并返回一个包含这两个部分的元组。这对于需要进一步处理文件名的情况非常有用。

import os
path = "/home/user/documents/report.pdf"
filename, extension = os.path.splitext(os.path.basename(path))
print("Filename:", filename)  # 输出：report
print("Extension:", extension)  # 输出：.pdf

二、PATHLIB 模块

pathlib 模块是Python 3.4引入的一个新模块，提供了面向对象的路径操作方式。它比os.path更加直观和易用。

from pathlib import Path
path = Path("/home/user/documents/report.pdf")
filename = path.name
print(filename)  # 输出：report.pdf

1、Path.name 属性

Path.name 属性会返回路径的最后一个部分，即文件名。如果路径指向一个目录，则返回目录名。

from pathlib import Path
def get_filename(path_str):
    path = Path(path_str)
    return path.name
print(get_filename("/home/user/documents/report.pdf"))  # 输出：report.pdf

2、Path.stem 属性

Path.stem 属性会返回路径的“纯名”部分，即不包括扩展名的文件名。

from pathlib import Path
path = Path("/home/user/documents/report.pdf")
filename_without_extension = path.stem
print(filename_without_extension)  # 输出：report

3、Path.suffix 属性

Path.suffix 属性会返回路径的扩展名部分。

from pathlib import Path
path = Path("/home/user/documents/report.pdf")
extension = path.suffix
print(extension)  # 输出：.pdf

三、OS.SCANDIR() 方法

os.scandir() 方法提供了一种高效的方式来遍历目录中的条目。它返回的每个条目都包含文件名和其他元数据。

1、使用os.scandir()获取文件名

import os
def get_filenames(directory):
    with os.scandir(directory) as entries:
        filenames = [entry.name for entry in entries if entry.is_file()]
    return filenames
print(get_filenames("/home/user/documents"))

2、结合os.path.basename()使用os.scandir()

你也可以结合 os.path.basename() 使用 os.scandir() 来获取文件名。

import os
def get_filenames(directory):
    with os.scandir(directory) as entries:
        filenames = [os.path.basename(entry.path) for entry in entries if entry.is_file()]
    return filenames
print(get_filenames("/home/user/documents"))

四、使用正则表达式获取文件名

虽然不推荐使用正则表达式来获取文件名，但在某些特定情况下，正则表达式可能会更灵活。

import re
def get_filename(path):
    pattern = r'[^\\/]+$'
    match = re.search(pattern, path)
    return match.group(0) if match else None
print(get_filename("/home/user/documents/report.pdf"))  # 输出：report.pdf

五、综合示例

为了帮助你更好地理解这些方法，我们将结合多个方法创建一个综合示例。

import os
from pathlib import Path
import re
def get_filename_with_os_path(path):
    return os.path.basename(path)
def get_filename_without_extension_with_os_path(path):
    return os.path.splitext(os.path.basename(path))[0]
def get_filename_with_pathlib(path):
    return Path(path).name
def get_filename_without_extension_with_pathlib(path):
    return Path(path).stem
def get_extension_with_pathlib(path):
    return Path(path).suffix
def get_filenames_in_directory_with_scandir(directory):
    with os.scandir(directory) as entries:
        return [entry.name for entry in entries if entry.is_file()]
def get_filename_with_regex(path):
    pattern = r'[^\\/]+$'
    match = re.search(pattern, path)
    return match.group(0) if match else None
path = "/home/user/documents/report.pdf"
directory = "/home/user/documents"
print("Using os.path.basename():", get_filename_with_os_path(path))
print("Using os.path.splitext():", get_filename_without_extension_with_os_path(path))
print("Using Path.name:", get_filename_with_pathlib(path))
print("Using Path.stem:", get_filename_without_extension_with_pathlib(path))
print("Using Path.suffix:", get_extension_with_pathlib(path))
print("Using os.scandir():", get_filenames_in_directory_with_scandir(directory))
print("Using regex:", get_filename_with_regex(path))