LuyangZ commited on
Commit
a0423ce
·
verified ·
1 Parent(s): 648d064

Update utils/clean_data.py

Browse files
Files changed (1) hide show
  1. utils/clean_data.py +6 -4
utils/clean_data.py CHANGED
@@ -60,6 +60,7 @@ def read_wall_wkt(wall, wkt):
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 = []
@@ -67,10 +68,11 @@ def read_wall_wkt(wall, wkt):
67
  if p not in remove_index:
68
  wkt_u = wkt_c[p]
69
  wkt_f.append(wkt_u)
70
-
71
- if p < (len(wkt_c)-1):
72
- wall_u = wall_c[p]
73
- wall_f.append(wall_u)
 
74
 
75
 
76
  if wall_num == rest_num:
 
60
  re_num = len(remove_index)
61
  rest_num = len(wkt_c) - re_num - 1
62
  wall_num = len(wall_c)
63
+ wkt_wall = len(wkt_c) - 1
64
 
65
  wall_f = []
66
  wkt_f = []
 
68
  if p not in remove_index:
69
  wkt_u = wkt_c[p]
70
  wkt_f.append(wkt_u)
71
+
72
+ if wall_num == wkt_wall:
73
+ if p < (len(wkt_c)-1):
74
+ wall_u = wall_c[p]
75
+ wall_f.append(wall_u)
76
 
77
 
78
  if wall_num == rest_num: