python 对shp进行掩膜

你可以使用Python中的geopandas库和rasterio库来对shp进行掩膜。

首先，你需要打开shp文件和用于掩膜的栅格文件。可以使用geopandas打开shp文件，使用rasterio打开栅格文件。然后，你需要将shp文件转换为与栅格文件相同的投影和空间参考系。

接下来，你可以使用geopandas中的sjoin函数将shp文件和栅格文件进行空间连接，这样你就可以得到每个shp要素与栅格像素之间的关系。然后，你可以使用rasterio中的mask函数将栅格文件根据shp文件进行掩膜。

下面是一个简单的示例代码：

import geopandas as gpd
import rasterio
from rasterio.mask import mask

# 打开shp文件
shp = gpd.read_file('path/to/shapefile.shp')

# 打开栅格文件
with rasterio.open('path/to/raster.tif') as src:
    # 将shp文件转换为栅格文件的投影和空间参考系
    shp = shp.to_crs(src.crs)

    # 进行空间连接
    joined = gpd.sjoin(shp, gpd.GeoDataFrame(geometry=[src.bounds]))

    # 获取栅格文件中与shp要素相交的部分
    if not joined.empty:
        shapes = joined['geometry'].tolist()
        data, transform = mask(src, shapes, crop=True)

        # 保存掩膜后的栅格文件
        with rasterio.open('path/to/masked_raster.tif', 'w', **src.profile) as dst:
            dst.write(data, 1)

注意，这只是一个简单的示例代码，你需要根据你的具体情况进行修改和调整。