如何将一个扩展名的一个目录下的所有文件合并到一个文件夹中

import os
import shutil

def merge_files_by_extension(source_dir, target_dir, extension):
    # 创建目标文件夹
    if not os.path.exists(target_dir):
        os.makedirs(target_dir)

    # 遍历源目录下的所有文件
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            # 检查文件扩展名是否符合要求
            if file.endswith(extension):
                # 构建源文件的完整路径
                source_file = os.path.join(root, file)
                # 构建目标文件的完整路径
                target_file = os.path.join(target_dir, file)
                # 移动或复制文件到目标文件夹
                shutil.move(source_file, target_file)  # 或使用shutil.copy(source_file, target_file)

# 示例用法
source_directory = "/path/to/source/directory"
target_directory = "/path/to/target/directory"
file_extension = ".txt"

merge_files_by_extension(source_directory, target_directory, file_extension)