Add slope, access etc. to pairs output

quantifyearth · Sep 11, 2023 · 9de6d35 · 9de6d35
1 parent 06a01ab
commit 9de6d35
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/methods/matching/find_pairs.py b/methods/matching/find_pairs.py
@@ -103,14 +103,14 @@ def find_match_iteration(
         match = filtered_s.iloc[min_index]
 
         results.append(
-            [k_row.lat, k_row.lng] + [k_row[x] for x in luc_columns] + \
-            [match.lat, match.lng] + [match[x] for x in luc_columns]
+            [k_row.lat, k_row.lng] + [k_row[x] for x in luc_columns + distance_columns] + \
+            [match.lat, match.lng] + [match[x] for x in luc_columns + distance_columns]
         )
 
     columns = ['k_lat', 'k_lng'] + \
-        [f'k_{x}' for x in luc_columns] + \
+        [f'k_{x}' for x in luc_columns + distance_columns] + \
         ['s_lat', 's_lng'] + \
-        [f's_{x}' for x in luc_columns]
+        [f's_{x}' for x in luc_columns + distance_columns]
 
     results_df = pd.DataFrame(results, columns=columns)
     results_df.to_parquet(os.path.join(output_folder, f'{idx_and_seed[1]}.parquet'))