如何复制包含特定扩展名文件的目录列表？

import os
import shutil

def copy_files_with_extension(source_dir, target_dir, extension):
    file_list = []
    
    # 遍历目标目录
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            # 检查文件扩展名是否匹配
            if file.endswith(extension):
                file_path = os.path.join(root, file)
                file_list.append(file_path)
    
    # 复制匹配的文件到目标目录
    for file_path in file_list:
        target_path = os.path.join(target_dir, os.path.relpath(file_path, source_dir))
        os.makedirs(os.path.dirname(target_path), exist_ok=True)
        shutil.copy2(file_path, target_path)
    
    return file_list

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

copied_files = copy_files_with_extension(source_directory, target_directory, file_extension)
print("Copied files:")
for file_path in copied_files:
    print(file_path)