LuyangZ commited on
Commit
5580c60
·
verified ·
1 Parent(s): 2995e6d

Update utils/clean_data.py

Browse files
Files changed (1) hide show
  1. 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 wall_f, wkt_f
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