python - 压缩属于同月的文件

Question

我有一个名为“DataFiles”的文件夹，其中包含如下文件:2016-04-26.csv, 2016-04-27.csv,2016-05-12.csv...我想压缩属于同一个月的 csv 文件...为此，我尝试了以下代码:

import datetime
import os
import zipfile

yearP= None
monthP = None

for csvfiles in os.listdir("DataFiles"):
    csvname, formata = csvfiles.split(".")
    year, month, date = csvname.split("-")
    if (yearP==year and monthP == month):
        zip = zipfile.ZipFile('%s.zip' %monthyear , 'a') 
        zip.write('%s.csv' %csvname ) 
        yearP= year
        monthP = month
    else:
        zip = zipfile.ZipFile('%s.zip' %monthyear , 'a') 
        zip.write('%s.csv' %csvname )   
        yearP= year
        monthP = month

但由于未按排序顺序返回异常结果。我想根据月份压缩并将压缩文件命名为相应的月份。

Answer 1

由于您的文件是 YYYY-MM-DD 格式，您可以按字典顺序对它们进行排序。

替换

for csvfiles in os.listdir("DataFiles"):

与

for csvfiles in sorted(os.listdir("DataFiles")):

---

对于大量 .csv 文件，这可能会变得不那么容易处理，您可能需要重新考虑您的算法，但对于少量文件，没关系。