Python 如何在Python中替换（或删除）文件名的扩展名

在本文中，我们将介绍如何在Python中替换（或删除）文件名的扩展名。当我们需要处理文件名时，有时需要改变文件的扩展名或者完全去掉扩展名。Python提供了一些简单而强大的方法来完成这些任务。

替换文件扩展名

在Python中，我们可以使用re模块来匹配和替换文件名的扩展名。re模块是Python标准库中的正则表达式模块，可以通过正则表达式模式匹配字符串并进行替换操作。

下面是一个示例代码，展示了如何替换文件名的扩展名：

import re

def replace_extension(filename, new_extension):
    # 使用正则表达式匹配文件名和扩展名
    pattern = r'(\w+)\.\w+'
    match = re.match(pattern, filename)

    if match:
        # 获取文件名和扩展名
        basename = match.group(1)

        # 替换扩展名
        new_filename = f"{basename}.{new_extension}"
        return new_filename
    else:
        # 如果文件名没有扩展名，则直接返回原文件名
        return filename

# 测试替换文件扩展名
filename = "example.txt"
new_extension = "csv"
new_filename = replace_extension(filename, new_extension)
print(new_filename)  # 输出: example.csv

在上面的示例中，我们定义了一个replace_extension函数，该函数接受文件名和新的扩展名作为参数。然后，我们使用正则表达式模式(\w+)\.\w+来匹配文件名和扩展名。如果匹配成功，我们提取出文件名并将新的扩展名拼接到文件名中，最后返回新的文件名。如果文件名没有扩展名，则直接返回原文件名。

删除文件扩展名

除了替换文件名的扩展名，有时我们也需要删除文件名的扩展名。在Python中，我们可以使用os.path模块提供的函数来完成这个任务。

下面是一个示例代码，展示了如何删除文件名的扩展名：

import os

def remove_extension(filename):
    # 使用os.path模块获取文件名和扩展名
    basename = os.path.splitext(filename)[0]

    # 返回文件名（不包含扩展名）
    return basename

# 测试删除文件扩展名
filename = "example.txt"
basename = remove_extension(filename)
print(basename)  # 输出: example

在上面的示例中，我们使用了os.path.splitext函数来获取文件名和扩展名。os.path.splitext函数将文件名分割成路径和扩展名两部分，并以元组的形式返回。我们通过索引0获取文件名（不包含扩展名）。