112 lines
4.4 KiB
Python
112 lines
4.4 KiB
Python
import geopandas as gpd
|
||
import pandas as pd
|
||
import rasterio
|
||
from rasterio.features import rasterize
|
||
import numpy as np
|
||
from shapely.geometry import mapping
|
||
|
||
|
||
def merge_and_rasterize_vectors_two_categories(file_paths, output_vector_path, output_raster_path,
|
||
raster_resolution=30):
|
||
"""
|
||
合并2个矢量图层（forest, water），转为栅格并显示2个类别，保留0值作为背景
|
||
|
||
参数:
|
||
file_paths: 包含2个矢量文件路径的列表 (按顺序: forest, water)
|
||
output_vector_path: 输出合并后的矢量文件路径
|
||
output_raster_path: 输出栅格文件路径
|
||
raster_resolution: 输出栅格的分辨率（默认30米）
|
||
"""
|
||
try:
|
||
# 检查输入文件数量
|
||
if len(file_paths) != 2:
|
||
raise ValueError("请提供正好2个矢量文件路径")
|
||
|
||
# 定义类别映射，按照指定顺序
|
||
categories = ['forest', 'water']
|
||
category_values = {'forest': 1, 'water': 2} # forest=1, water=2
|
||
|
||
# 读取所有矢量文件并添加分类
|
||
gdfs = []
|
||
for i, (path, category) in enumerate(zip(file_paths, categories), 1):
|
||
print(f"正在读取第{i}个矢量文件: {path}")
|
||
gdf = gpd.read_file(path)
|
||
gdf['category'] = category # 添加类别字段
|
||
gdf['value'] = category_values[category] # 添加数值字段用于栅格化
|
||
gdfs.append(gdf)
|
||
|
||
# 检查和统一坐标系（以第一个图层为基准）
|
||
base_crs = gdfs[0].crs
|
||
for i, gdf in enumerate(gdfs[1:], 2):
|
||
if gdf.crs != base_crs:
|
||
print(f"第{i}个图层坐标系不同，正在转换为第一个图层的坐标系...")
|
||
gdfs[i - 1] = gdf.to_crs(base_crs)
|
||
|
||
# 合并所有GeoDataFrame
|
||
print("正在合并图层...")
|
||
merged_gdf = gpd.GeoDataFrame(pd.concat(gdfs, ignore_index=True))
|
||
merged_gdf = merged_gdf.set_geometry('geometry')
|
||
|
||
# 保存合并后的矢量文件
|
||
print(f"正在保存合并矢量结果到: {output_vector_path}")
|
||
merged_gdf.to_file(output_vector_path)
|
||
|
||
# 计算栅格化的范围
|
||
bounds = merged_gdf.total_bounds # [minx, miny, maxx, maxy]
|
||
width = int((bounds[2] - bounds[0]) / raster_resolution)
|
||
height = int((bounds[3] - bounds[1]) / raster_resolution)
|
||
|
||
# 创建栅格化变换
|
||
transform = rasterio.transform.from_bounds(
|
||
bounds[0], bounds[1], bounds[2], bounds[3], width, height
|
||
)
|
||
|
||
# 栅格化：将几何对象转为栅格，使用'value'字段作为像素值
|
||
print("正在将矢量转为栅格...")
|
||
shapes = ((mapping(geom), value) for geom, value in zip(merged_gdf.geometry, merged_gdf['value']))
|
||
raster = rasterize(
|
||
shapes=shapes,
|
||
out_shape=(height, width),
|
||
transform=transform,
|
||
fill=0, # 背景值为0
|
||
dtype=rasterio.uint8
|
||
)
|
||
|
||
# 保存栅格文件
|
||
with rasterio.open(
|
||
output_raster_path,
|
||
'w',
|
||
driver='GTiff',
|
||
height=height,
|
||
width=width,
|
||
count=1,
|
||
dtype=rasterio.uint8,
|
||
crs=base_crs,
|
||
transform=transform,
|
||
nodata=0 # 设置nodata值为0，表示背景
|
||
) as dst:
|
||
dst.write(raster, 1)
|
||
|
||
print("处理完成！")
|
||
print(f"合并矢量保存为: {output_vector_path}")
|
||
print(f"栅格保存为: {output_raster_path}")
|
||
print(f"栅格中类别值: 0=背景, 1=forest, 2=water")
|
||
|
||
except Exception as e:
|
||
print(f"发生错误: {str(e)}")
|
||
|
||
|
||
# 使用示例
|
||
if __name__ == "__main__":
|
||
# 输入2个矢量文件路径（按顺序: forest, water）
|
||
input_files = [
|
||
r"E:\Data\z18\sd\Total_sd\shandonghebingshp\sdshp\forest\sdforest.shp",
|
||
r"E:\Data\z18\sd\Total_sd\shandonghebingshp\sdshp\water\true\shandongsuiyu3.shp"
|
||
]
|
||
output_vector_file = r"E:\njds\shenyang\shengyangshp\kejianshe\merged_vector_forest_water.shp"
|
||
output_raster_file = r"E:\njds\shenyang\shengyangshp\kejianshe\merged_raster_forest_water.tif"
|
||
|
||
# 执行合并和栅格化
|
||
merge_and_rasterize_vectors_two_categories(input_files, output_vector_file, output_raster_file,
|
||
raster_resolution=10)
|