Skip to content

Commit

Permalink
removes route filtering
Browse files Browse the repository at this point in the history
  • Loading branch information
rjhanes committed Sep 15, 2021
1 parent 2909787 commit 11f206f
Showing 1 changed file with 2 additions and 13 deletions.
15 changes: 2 additions & 13 deletions celavi/data_filtering.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@



def data_filter(loc_file_name,routes_file_name,num_of_turbines_filename,states):
def data_filter(loc_file_name,num_of_turbines_filename,states):

locations = pd.read_csv(loc_file_name)

Expand All @@ -13,18 +13,7 @@ def data_filter(loc_file_name,routes_file_name,num_of_turbines_filename,states):

facililites_included = locations_filtered[['facility_id', 'facility_type', 'lat', 'long']]
facility_id_included = list(pd.unique(facililites_included['facility_id']))



routes = pd.read_csv(routes_file_name)
routes_filtered = routes[(routes['source_facility_id'].isin(facility_id_included)) &
(routes['destination_facility_id'].isin(facility_id_included))]
routes_filtered = routes_filtered.drop_duplicates()
routes_filtered.to_csv(routes_file_name)





number_of_turbines = pd.read_csv(num_of_turbines_filename)
number_of_turbines_filtered = number_of_turbines[number_of_turbines['facility_id'].isin(facility_id_included)]
number_of_turbines_filtered.to_csv(num_of_turbines_filename)

0 comments on commit 11f206f

Please sign in to comment.