indices to remove False doesn't work

Author: tiffanychu90

rt_segment_speeds/scripts/model_speeds_every100.py

@@ -47,7 +47,7 @@ def grab_arrays_by_trip(df, meters_interval: int):
 def get_speeds_every_interval(
     one_trip_distance_arr, 
     one_trip_timestamp_arr,
-    intervaled_distance_cutoffs
+    intervaled_distance_cutoffs,
 ):
 
     one_trip_speed_series = []
@@ -87,6 +87,8 @@ def grab_arrays_by_trip2(
 
         one_trip_distance_arr = getattr(row, "interpolated_distances")
         one_trip_timestamp_arr = getattr(row, "resampled_timestamps")
+        should_calculate = np.array(getattr(row, "stop_meters_increasing"))
+
 
         start_dist = int(np.floor(one_trip_distance_arr).min())
         end_dist = int(np.ceil(one_trip_distance_arr).max())        
@@ -97,26 +99,32 @@ def grab_arrays_by_trip2(
 
         elif distance_type == "stop_to_stop":
             intervaled_distance_cutoffs = getattr(row, intervaled_distance_column_or_meters)
-        
+            #do_not_calculate_indices = np.where(should_calculate == False)[0]
+
         speeds_for_trip = get_speeds_every_interval(
             one_trip_distance_arr, 
             one_trip_timestamp_arr,
-            intervaled_distance_cutoffs
+            intervaled_distance_cutoffs,
         )
-        speed_series.append(speeds_for_trip)
-
+        
+        #if len(do_not_calculate_indices) > 0:
+        #    speeds_for_trip[do_not_calculate_indices] = np.nan
+        
 
         if distance_type == "equal_intervals":
             intervaled_cutoffs.append(intervaled_distance_cutoffs)
             keep_cols = ["intervaled_meters", "speeds"]
         elif distance_type == "stop_to_stop":
             keep_cols = ["speeds", "stop_sequence"]
-    
+        
+        speed_series.append(speeds_for_trip)
+
     if distance_type == "equal_intervals":
         df2 = df.assign(
             intervaled_meters = intervaled_cutoffs,
             speeds = speed_series
         )
+        
     elif distance_type == "stop_to_stop":
         df2 = df.assign(
             speeds = speed_series
@@ -166,7 +174,7 @@ def grab_arrays_by_trip2(
         stop_time_cutoffs = pd.read_parquet(
             f"{SEGMENT_GCS}stop_times_projected_{analysis_date}.parquet",
             filters = [[("trip_instance_key", "in", subset_trips)]],
-            columns = ["trip_instance_key", "stop_sequence", "stop_meters"]
+            columns = ["trip_instance_key", "stop_sequence", "stop_meters", "stop_meters_increasing"]
         )
 
         gdf = pd.merge(
@@ -179,7 +187,7 @@ def grab_arrays_by_trip2(
         results = grab_arrays_by_trip2(
             gdf,
             distance_type = "stop_to_stop",
-            intervaled_distance_column_or_meters = "stop_meters"
+            intervaled_distance_column_or_meters = "stop_meters",
         )
 
         results.to_parquet(