Spaces:
Runtime error
Runtime error
Update utils/clean_data.py
Browse files- utils/clean_data.py +10 -1
utils/clean_data.py
CHANGED
|
@@ -57,6 +57,10 @@ def read_wall_wkt(wall, wkt):
|
|
| 57 |
ri = dup_index[j]
|
| 58 |
remove_index.append(ri)
|
| 59 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 60 |
wall_f = []
|
| 61 |
wkt_f = []
|
| 62 |
for p in range(len(wkt_c)):
|
|
@@ -68,10 +72,15 @@ def read_wall_wkt(wall, wkt):
|
|
| 68 |
wall_u = wall_c[p]
|
| 69 |
wall_f.append(wall_u)
|
| 70 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 71 |
wkt_f = ", ".join(wkt_f)
|
| 72 |
wkt_f = "POLYGON ((" + wkt_f + "))"
|
| 73 |
|
| 74 |
-
return
|
| 75 |
|
| 76 |
|
| 77 |
|
|
|
|
| 57 |
ri = dup_index[j]
|
| 58 |
remove_index.append(ri)
|
| 59 |
|
| 60 |
+
re_num = len(remove_index)
|
| 61 |
+
rest_num = len(wkt_c) - re_num - 1
|
| 62 |
+
wall_num = len(wall_c)
|
| 63 |
+
|
| 64 |
wall_f = []
|
| 65 |
wkt_f = []
|
| 66 |
for p in range(len(wkt_c)):
|
|
|
|
| 72 |
wall_u = wall_c[p]
|
| 73 |
wall_f.append(wall_u)
|
| 74 |
|
| 75 |
+
if wall_num == rest_num:
|
| 76 |
+
wall_ff = wall_c
|
| 77 |
+
else:
|
| 78 |
+
wall_ff = wall_f
|
| 79 |
+
|
| 80 |
wkt_f = ", ".join(wkt_f)
|
| 81 |
wkt_f = "POLYGON ((" + wkt_f + "))"
|
| 82 |
|
| 83 |
+
return wall_ff, wkt_f
|
| 84 |
|
| 85 |
|
| 86 |
|