feat: 🔗 I Am Speed

This commit is contained in:
2025-11-11 21:32:16 +00:00
parent a40e6d2872
commit be2c89bcc2
6 changed files with 50 additions and 51 deletions
+6 -8
View File
@@ -1,5 +1,6 @@
import pandas as pd
class CombineTimeseries:
def __init__(self, config, locations):
self.config = config
@@ -7,7 +8,6 @@ class CombineTimeseries:
self.grouped_locations = {}
self.build_location_groups()
def build_location_groups(self):
for location in self.locations:
group = location[4] # output group is at index 4
@@ -15,19 +15,17 @@ class CombineTimeseries:
self.grouped_locations[group] = []
self.grouped_locations[group].append(location)
def combine_csv_files(self):
for group, loc_list in self.grouped_locations.items():
combined_df = None
for loc in loc_list:
csv_to_load = f'./csv_files/{loc[0]}_timeseries_data.csv'
csv_to_load = f"./csv_files/{loc[0]}_timeseries_data.csv"
df = pd.read_csv(csv_to_load, index_col=0)
if combined_df is None:
combined_df = df
else:
combined_df = combined_df.join(df, how='inner')
output_file = f'{self.config.COMBINED_FOLDER}/group_{group}_timeseries_data.csv'
combined_df = combined_df.join(df, how="inner")
output_file = (
f"{self.config.COMBINED_FOLDER}/group_{group}_timeseries_data.csv"
)
combined_df.to_csv(output_file)