From 0e534cd490327369d814e33d91422cb91aa6156a Mon Sep 17 00:00:00 2001
From: baol <j2eer@163.com>
Date: Fri, 29 Nov 2024 10:38:30 +0800
Subject: [PATCH] add feature

---
 main.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/main.py b/main.py
index a6644a4..2fd4242 100644
--- a/main.py
+++ b/main.py
@@ -127,7 +127,7 @@ def generate_river_czml(river_df, runoff_df, level=4, start_time=None):
 
     # 为每条河段创建一个entity
     for idx in river_df.index:
-        print(idx)
+        river_index = river_df.loc[idx, "Index"]
         coords = mapping(river_df.loc[idx, "geometry"])["coordinates"]
         # new_coords = [coord for point in coords for coord in point]
         # print(new_coords)
@@ -150,7 +150,7 @@ def generate_river_czml(river_df, runoff_df, level=4, start_time=None):
 
         # 创建河段实体
         entity = {
-            "id": f"river_{idx}",
+            "id": f"{river_index}",
             "name": f"River Segment {idx}",
             "polyline": {
                 "positions": {