diff --git a/my_utils.py b/my_utils.py index 923f010..72a7d92 100644 --- a/my_utils.py +++ b/my_utils.py @@ -1,2 +1,10 @@ def get_column(file_name, query_column, query_value, result_column): - return None + results =[] + + with open(file_name,'r') as file: + for line in file: + values=line.split(',') + if values[query_column]==query_value: + results.append(values[result_column]) + + return results diff --git a/print_fires.py b/print_fires.py index 4992fb6..fa2d670 100644 --- a/print_fires.py +++ b/print_fires.py @@ -1,6 +1,9 @@ +import my_utils + country='United States of America' -county_column = 1 -fires_column = 4 +county_column = 0 +fires_column = 3 file_name = 'Agrofood_co2_emission.csv' -fires = get_column() +fires = my_utils.get_column(file_name,county_column,country,fires_column) + print(fires)