如何压缩指定后缀（.npy/.csv）文件？代码仅压缩单个文件问题排查

原代码的核心问题如下：

• 缩进错误：压缩包创建逻辑嵌套在文件遍历循环里，每次循环都会重新生成压缩包，覆盖之前的内容，导致最终只有单个文件
• 文件收集逻辑错误：list_files = (file)每次都会覆盖之前的文件，没有做追加，而且误用元组存储文件列表
• 路径拼接错误：直接用'\'拼接路径容易引发转义问题，且当list_files是元组时会触发TypeError
• 冗余嵌套循环：手动四层遍历目录太繁琐，可通过os.walk简化
• 语法错误：用位运算符|替代了逻辑运算符or，判断文件后缀时会出错
• 多余的关闭操作：with语句会自动关闭压缩包，无需手动调用zipMe.close()

修正后的代码 #

import os
import zipfile

# 根目录
root_dir = os.path.join(os.getcwd(), 'recs')
target_folder_name = 'RadarIfxAvian'

# 遍历所有目录
for dirpath, dirnames, filenames in os.walk(root_dir):
    # 检查当前目录路径是否包含目标文件夹名
    if target_folder_name in dirpath:
        # 定义压缩包路径：放在目标目录下，命名为radar.zip
        zip_path = os.path.join(dirpath, 'radar.zip')
        # 创建压缩包，使用ZIP_DEFLATED压缩
        with zipfile.ZipFile(zip_path, 'w', compression=zipfile.ZIP_DEFLATED) as zip_me:
            # 遍历当前目录下的所有文件
            for filename in filenames:
                # 筛选.npy和.csv后缀的文件
                if filename.endswith('.npy') or filename.endswith('.csv'):
                    # 拼接文件的完整路径
                    file_full_path = os.path.join(dirpath, filename)
                    # 将文件添加到压缩包，arcname参数保持文件原名称，避免压缩包内出现完整路径层级
                    zip_me.write(file_full_path, arcname=filename)
        print(f"已完成压缩：{zip_path}")

关键修改说明 #

1. 用os.walk简化遍历：自动递归遍历所有子目录，无需手动多层嵌套循环
2. 正确收集并添加文件：在压缩包的with块内遍历筛选文件，一次性添加所有符合条件的文件到同一个压缩包
3. 安全路径拼接：全程使用os.path.join处理路径，避免转义错误和系统兼容性问题
4. 修复逻辑运算符：用or替代|，正确判断文件后缀
5. 优化压缩包内路径：通过arcname参数，让压缩包内的文件保持原名称，不会带上完整的目录路径

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