Создание отдельных архивов из отдельных папок с ZipFile

Question

У меня есть каталог, состоящий из 3-х папок:
----- 001 -----
----- 003

I» d хотел бы создать 3 архива с именем 001.zip, 002.zip и 003.zip. Каждый архив должен состоять из содержимого папки.

Я использую ZipFile библиотеку, и мне удалось сделать архив одной папки:

import os, zipfile  
zf = zipfile.ZipFile('C:/zipfile4.zip', mode='w') 
for dirpath,dirs,files in os.walk("C:/Test/002"): 
    for f in files: 
     fn = os.path.join(dirpath, f) 
     zf.write(fn) 

Но я не знаю, как сделать, чтобы создать много архивов подряд, используя эту ZipFile библиотеку.
Не могли бы вы мне помочь?
Спасибо!

Answer 1

Использование образца кода в качестве основы, это должно работать:

import os 
import zipfile 

dir = "C:/Test/002" 

# get list of all subdirectories in the directory 
subdirs = [subdir for subdir in os.listdir(dir) if os.path.isdir(os.path.join(dir, subdir))] 

# create a zip archive for each subdirectory 
for subdir in subdirs: 
    zf_name = "C:/" + subdir + ".zip" 
    zf = zipfile.ZipFile(zf_name, mode='w') 
    for dirpath,dirs,files in os.walk(os.path.join(dir, subdir)): 
     for f in files: 
      fn = os.path.join(dirpath, f) 
      zf.write(fn)