使用python的geopandas库实现shp转栅格

要使用geopandas库实现shp转栅格，您需要首先安装geopandas库。您可以使用以下命令来安装geopandas：

pip install geopandas

接下来，您需要准备一个shp文件。确保您已经将shp文件放置在您的工作目录中。

下面是一个示例代码，演示了如何使用geopandas库将shp文件转换为栅格数据：

import geopandas as gpd
from rasterio.plot import show
from rasterio.transform import from_origin
from rasterio import Affine

# 读取shp文件
shp_file = 'path/to/your/shp_file.shp'
gdf = gpd.read_file(shp_file)

# 获取shp文件的边界框
bbox = gdf.bounds

# 设置栅格的分辨率和大小
resolution = 0.01
width = int((bbox.maxx - bbox.minx) / resolution)
height = int((bbox.maxy - bbox.miny) / resolution)

# 创建栅格数组
grid = np.zeros((height, width), dtype=np.uint8)

# 将shp文件中的几何形状转换为栅格数据
for shape in gdf.geometry:
    coords = np.array(shape.exterior.coords)
    x = ((coords[:, 0] - bbox.minx) / resolution).astype(int)
    y = ((coords[:, 1] - bbox.miny) / resolution).astype(int)
    grid[y, x] = 1

# 创建栅格文件
transform = from_origin(bbox.minx, bbox.maxy, resolution, resolution)
affine = Affine.from_gdal(*transform)
with rasterio.open('path/to/your/output/raster.tif', 'w', driver='GTiff', height=height, width=width, count=1, dtype=np.uint8, crs=gdf.crs, transform=affine) as dst:
    dst.write(grid, 1)

# 可选：显示栅格数据
with rasterio.open('path/to/your/output/raster.tif') as src:
    show(src)

请注意，这只是一个示例代码，您需要根据您的shp文件的特定要求进行调整。确保将path/to/your/shp_file.shp替换为您的shp文件的路径，将path/to/your/output/raster.tif替换为您希望将栅格数据保存到的路径。

希望这可以帮助到您