矢量文件剪裁矢量文件（Python）

# encoding: utf-8 ''' @File : Shp_clip_shp_gdal.py @Time : 2022/03/26 18:27:14 @Author : Qi Cheng Yi Shao @Version : 1.0 @Contact : 1441210996@qq.com @License : (C)Copyright 2021-2024, Mutual Aid Commune @Desc : None ''' import os import numpy as np import geopandas as gpd import warnings import argparse import glob from tqdm import tqdm parser = argparse.ArgumentParser(description="矢量裁剪矢量") parser.add_argument('--shp_path', type=str,default="label/")#待裁剪的矢量文件的文件夹 parser.add_argument('--boundary_path', type=str,default="boundary/clip.shp")#边框矢量文件 parser.add_argument('--result_dir', type=str,default="label_clip/")#剪裁后矢量文件文件夹 warnings.filterwarnings('ignore', 'GeoSeries.notna', UserWarning) def shp_clip_shp(source_shp,mask_shp,result_shp): shp_gpd=gpd.read_file(source_shp) mask_gpd=gpd.read_file(mask_shp) mask_clip=gpd.clip(shp_gpd,mask_gpd) mask_clip=mask_clip[~mask_clip.is_empty] mask_clip.to_file(result_shp,encoding='utf-8') if __name__=="__main__": args = parser.parse_args() shp_list = glob.glob(os.path.join(args.shp_path, "*.shp")) for path in tqdm(shp_list): shp_name = os.path.basename(path) # print path shp_clip_shp(path,args.boundary_path, os.path.join(args.result_dir, shp_name))