Python复制矢量,按日期批量复制,并形成压缩包
1.导入所需要的库
import geopandas as gpd
import datetime
from datetime import datetime, timedelta
import glob
import os
import geopandas as gpd
import zipfile
import openpyxl
import pandas as pd
import numpy as np
import osgeo.ogr , osgeo.gdal
from osgeo import ogr,gdal,osr
2.定义一个形成时间序列的函数
def time_minmax(start_date,end_date):
date_list = []
current_date = start_date
while current_date <= end_date:
# 将日期格式化为“年月日”并添加到列表中
date_str = current_date.strftime("%Y%m%d")
date_list.append(date_str)
# 增加一天
current_date += timedelta(days=1)
# 打印生成的日期列表
return date_list
3.主函数
if __name__ == "__main__":
# 读取原始的shapefile
input_shp = r"Your_Path\\test.shp" # 替换为你的输入文件路径
out_path = r'Your_Path\\OutPut_Path\\'
# date = (datetime.datetime.now()).strftime('%Y%m%d')
# name = date[:]
### 定义起止时间
start_date = datetime(2024, 1, 6) #复制为新矢量的开始时间(按实际修改)
end_date = datetime(2024, 1, 9) #复制为新矢量的结束时间(按实际修改)
data_list = time_minmax(start_date , end_date) ### 生成列表
original_gdf = gpd.read_file(input_shp)
# 将数据写入新的shapefile
for i in range(len(data_list)):
out_path_test = os.path.join(out_path, data_list[i])
if os.path.exists(out_path_test):
pass
else:
os.makedirs(out_path_test)
output_shp = os.path.join(out_path_test, f"{data_list[i]}_test.shp") # 替换为你想要的输出文件路径
output_gdf = original_gdf.copy() # 复制原始的 GeoDataFrame
original_gdf.to_file(output_shp)
# zip文件名称
zip_file = f"{out_path_test}" + "_test.zip"
# 压缩文件夹内容为ZIP
with zipfile.ZipFile(zip_file, 'w', zipfile.ZIP_DEFLATED) as zipf:
for root, dirs, files in os.walk(out_path_test):
for file in files:
zipf.write(os.path.join(root, file), os.path.relpath(os.path.join(root, file), out_path_test))