shpfile文件根据mask掩码切割相同大小的文件

shpfile文件根据mask掩码切割

    • 1.导包
    • 2.相关代码

1.导包

import pyproj
import pandas as pd
import geopandas as gpd

2.相关代码

  • grid文件是网格数据,在qgis中生成获得
shp=gpd.read_file('a.shp',encoding='utf-8')
shp_2022_4.to_crs(epsg=4549, inplace=True)# 设置投影方式
# 读取grid文件
grid=gpd.read_file('grid.shp',encoding='utf-8')
grid.to_crs(epsg=4549, inplace=True)

for i in range(len(grid)):
    clip=gpd.clip(shp,grid.iloc[i,1],keep_geom_type =True)# 切割
    if len(clip.index)==0: # 切割范围没有数据
        continue
    df=pd.DataFrame()  # 创建dataframe对象
    grid_data=pd.DataFrame([len(df),0,0,grid.iloc[i,1]]).T # 获取grid网格边框数据,确保大小一致
    df=df.append(grid_data)
    long=len(df)
    for j in range(len(clip)): 
        geo=clip.iloc[j,-1]
        pre=pre_type[clip.iloc[j,6]]
        back=after_type[clip.iloc[j,7]]
        clip_data=pd.DataFrame([long,pre,back,geo]).T # 获取shp文件需要的数据
        df=df.append(clip_data)
        long=long+1
    df.columns=['id','pre_type','back_type','geometry']
    df[['pre_type','back_type']]=df[['pre_type','back_type']].astype(object)
    data=gpd.GeoDataFrame(df) # dataframe to GeoDataFrame
    data.crs = pyproj.CRS.from_user_input('EPSG:4549') # 设置投影
    name=os.path.join(path_shp,f'clip_{i}.shp')
    data.to_file(name, driver='ESRI Shapefile',encoding='utf-8')  # 保存切割的文件  

你可能感兴趣的:(其他,python,shp文件切割,shp文件根据mask掩码切割,geopandas.clip,shpfile文件操作)