Python按条件复制文件问题：匹配4位ID时复制异常或无结果

我来帮你搞定这个文件复制的问题！你现在的代码主要有几个小问题导致复制出错或者没反应，我给你梳理下并修正：

问题根源 #

• 你的ok_ids里是整数类型，但文件名开头是字符串，直接对比会完全不匹配
• 代码里files[:...没写完，应该取文件名前4个字符来匹配4位ID
• 重复的ID会导致多次无效判断，既浪费资源又可能重复复制
• os.listdir返回的只是文件名，不是完整路径，直接用shutil.copy会找不到文件
• 没判断条目是不是文件，可能会不小心复制文件夹

修正后的完整代码 #

import shutil
import os

# 把ID转成字符串并去重，避免类型不匹配和重复判断
ok_ids = {str(id) for id in [5252, 8396, 8397, 8397, 8556, 8004, 6545, 6541, 4392, 4392, 6548, 1363, 1363, 1363, 8489, 8652, 1368, 1368]}
source_dir = "/Users/amm/Desktop/mypath1/"
destination_dir = "/Users/amm/Desktop/mypath2/"

# 遍历源目录下的所有条目
for filename in os.listdir(source_dir):
    # 拼接完整的源文件路径，避免找不到文件
    source_path = os.path.join(source_dir, filename)
    # 只处理文件，跳过文件夹
    if not os.path.isfile(source_path):
        continue
    # 先判断文件名长度足够，避免索引错误
    if len(filename) >= 4:
        file_id = filename[:4]
        if file_id in ok_ids:
            # 拼接目标路径
            destination_path = os.path.join(destination_dir, filename)
            # 执行复制
            shutil.copy(source_path, destination_path)
            print(f"已成功复制: {filename}")

关键修正说明 #

• 转成字符串集合：既解决了整数和字符串的类型不匹配问题，又自动去重，让ID判断更高效
• 完整路径拼接：必须用os.path.join把目录和文件名拼起来，否则程序找不到要复制的文件
• 文件类型判断：确保只复制文件，不会把源目录里的文件夹也复制过去
• 长度校验：防止遇到文件名不足4位的情况，避免抛出索引错误
• 日志打印：方便你直观看到哪些文件被成功复制，排查问题更方便

内容的提问来源于stack exchange，提问作者Abbie