如何查找任何匹配子文件夹组中的所有文件并将其复制到新目录？

import os
import shutil

def copy_files_matching_pattern(source_dir, target_dir, pattern):
    # 遍历主文件夹及其所有子文件夹
    for root, dirs, files in os.walk(source_dir):
        # 在每个文件夹中查找匹配的文件
        for file in files:
            if pattern in file:
                # 构建源文件路径和目标文件路径
                source_file = os.path.join(root, file)
                target_file = os.path.join(target_dir, file)
                # 复制文件到新目录
                shutil.copy2(source_file, target_file)

# 示例用法
source_directory = '/path/to/source/directory'
target_directory = '/path/to/target/directory'
file_pattern = '.txt'  # 匹配以".txt"结尾的文件

copy_files_matching_pattern(source_directory, target_directory, file_pattern)