以下是一个完整的 Python 代码示例，用于将建筑物底面的 `shp` 数据拉升生成白膜模型，并导出为 `obj` 格式，包含 `main` 函数的调用实现。 ```python import geopandas as gpd import os def write_obj(vertices, faces, filename): with open(filename, 'w') as f: for v in vertices: f.write(f"v {v[0]} {v[1]} {v[2]}\n") for face in faces: f.write(f"f {' '.join(str(i) for i in face)}\n") def extrude_polygon(polygon_coords, height=10): bottom_vertices = [(x, y, 0) for x, y in polygon_coords] top_vertices = [(x, y, height) for x, y in polygon_coords] vertices = bottom_vertices + top_vertices n = len(polygon_coords) bottom_face = [i + 1 for i in range(n)] top_face = [i + 1 + n for i in range(n)] faces = [bottom_face, top_face] for i in range(n): next_i = (i + 1) % n faces.append([i + 1, next_i + 1, next_i + 1 + n, i + 1 + n]) return vertices, faces def process_shp_to_obj(shp_path, output_dir, height=10): gdf = gpd.read_file(shp_path) base_name = os.path.splitext(os.path.basename(shp_path))[0] for idx, row in gdf.iterrows(): geom = row['geometry'] if geom.geom_type == 'Polygon': coords = list(geom.exterior.coords) vertices, faces = extrude_polygon(coords, height) obj_path = os.path.join(output_dir, f"{base_name}_{idx}.obj") write_obj(vertices, faces, obj_path) def main(): input_dir = "shp_files" output_dir = "obj_models" os.makedirs(output_dir, exist_ok=True) for filename in os.listdir(input_dir): if filename.endswith(".shp"): shp_path = os.path.join(input_dir, filename) process_shp_to_obj(shp_path, output_dir) if __name__ == "__main__": main() ``` ### 说明 - 使用 `geopandas` 读取 `shp` 文件并提取建筑物底面轮廓。 - 将多边形数据进行三维拉伸，生成顶部和底部面，并连接侧面以形成封闭的三维模型。 - 将生成的模型数据写入 `.obj` 文件格式。 - 支持批量处理多个 `shp` 文件，并为每个建筑物生成独立的 `.obj` 模型文件。 - 确保模型的空间位置保持一致，适用于 GIS 数据的三维建模需求[^1]。 ###