seb/group-1-db23
Archived
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Load accident file from internet if not present.

Browse Source
This commit is contained in:
Sebastian Lenzlinger 2023-12-03 12:35:38 +01:00
parent 0afab39397
commit 65bd9480e7
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/integrate.py
View File

@ -65,8 +65,9 @@ def process_miv_data():
return cleaned_miv_df return cleaned_miv_df
def process_accident_data(): def process_accident_data(file_present: bool):
if not file_present:
du.process_urls(data_dir, accident_file_url)
acc_df_unified = du.load_dataframes_from_geojson_files(data_dir, accident_file_u_string) acc_df_unified = du.load_dataframes_from_geojson_files(data_dir, accident_file_u_string)
acc_cols_to_keep = ['AccidentUID', 'AccidentHour', 'AccidentYear', 'AccidentWeekDay_en', 'AccidentType', acc_cols_to_keep = ['AccidentUID', 'AccidentHour', 'AccidentYear', 'AccidentWeekDay_en', 'AccidentType',
'AccidentSeverityCategory', 'AccidentInvolvingPedestrian', 'AccidentInvolvingBicycle', 'AccidentSeverityCategory', 'AccidentInvolvingPedestrian', 'AccidentInvolvingBicycle',
@ -77,6 +78,6 @@ def process_accident_data():
if __name__ == '__main__': if __name__ == '__main__':
acc_df = process_accident_data() acc_df = process_accident_data(False)
print(acc_df.dtypes) print(acc_df.dtypes)
print(acc_df.head(100)) print(acc_df.head(100))