将特定文件从目录和子目录复制到新目录Python

import os
import shutil

def copy_files(source_dir, target_dir, file_extension):
    for root, dirs, files in os.walk(source_dir):
        for file in files:
            if file.endswith(file_extension):
                source_file = os.path.join(root, file)
                target_file = os.path.join(target_dir, file)
                shutil.copy2(source_file, target_file)

# 示例调用
source_dir = '/path/to/source/dir'
target_dir = '/path/to/target/dir'
file_extension = '.txt'  # 指定要复制的文件扩展名，例如.txt文件
copy_files(source_dir, target_dir, file_extension)