python获取文件后缀名及批量更新目录下文件后缀名的方法

下面是详细讲解“Python获取文件后缀名及批量更新目录下文件后缀名的方法”的完整攻略。

需求分析

在进行文件操作时，经常需要获取文件的后缀名，这样才能区分出不同类型的文件，并进行相应的操作。同时，有时我们也需要批量更改一个目录下所有文件的后缀名。本文将详细讲解这两个需求的实现方法。

获取文件后缀名

我们可以使用Python内置模块os和os.path来获取文件的后缀名。

import os

file_name = 'example.txt'
ext = os.path.splitext(file_name)[-1].lower()
print(ext)  # .txt

其中，os.path.splitext()函数可以将文件名和后缀名分开，[-1]表示获取后缀名部分，lower()方法将后缀名转换为小写字母形式。

批量更改文件后缀名

我们可以使用Python内置模块os和shutil来批量操作目录中的文件，并将它们的后缀名更改为目标后缀名。

import os
import shutil

source_dir = 'source_dir/'
target_dir = 'target_dir/'
source_ext = '.txt'
target_ext = '.md'

# 逐个处理文件
for file_name in os.listdir(source_dir):
    if not file_name.endswith(source_ext):
        continue
    # 构造新文件名，更改后缀名
    new_name = os.path.splitext(file_name)[0] + target_ext
    source_path = os.path.join(source_dir, file_name)
    target_path = os.path.join(target_dir, new_name)
    # 复制到目标目录并更名
    shutil.copy(source_path, target_path)

上述代码首先定义了源目录(source_dir)、目标目录(target_dir)、源后缀名(source_ext)和目标后缀名(target_ext)。然后使用os.listdir()函数逐个获取源目录下的文件名，并使用endswith()方法判断该文件的后缀名是否与源后缀名相同。如果不同，则continue跳过，否则根据目标后缀名构造新的文件名，并使用os.path.join()函数构造源文件路径(source_path)和目标文件路径(target_path)。最后使用shutil.copy()函数将源文件复制到目标文件，并更名为新的文件名。

示例说明

假设我们有一个目录example，其中存储了若干.txt格式的文件。现在需要将这些文件的后缀名统一更改为.md。

1. 获取文件后缀名

import os

file_name = 'example.txt'
ext = os.path.splitext(file_name)[-1].lower()
print(ext)  # .txt

运行结果为".txt"。

1. 批量更改文件后缀名

import os
import shutil

source_dir = 'example/'
target_dir = 'example_copy/'
source_ext = '.txt'
target_ext = '.md'

# 逐个处理文件
for file_name in os.listdir(source_dir):
    if not file_name.endswith(source_ext):
        continue
    # 构造新文件名，更改后缀名
    new_name = os.path.splitext(file_name)[0] + target_ext
    source_path = os.path.join(source_dir, file_name)
    target_path = os.path.join(target_dir, new_name)
    # 复制到目标目录并更名
    shutil.copy(source_path, target_path)

运行后，example_copy目录中的所有文件后缀名都将更改为.md。

以上就是Python获取文件后缀名及批量更新目录下文件后缀名的方法的完整攻略了。希望对你有所帮助。