diff --git a/my_utils.py b/my_utils.py index 923f010..9ed0d35 100644 --- a/my_utils.py +++ b/my_utils.py @@ -1,2 +1,11 @@ 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..5522c89 100644 --- a/print_fires.py +++ b/print_fires.py @@ -1,6 +1,9 @@ -country='United States of America' -county_column = 1 -fires_column = 4 +import my_utils + +country ='United States of America' +country_column = 0 +fires_column = 3 file_name = 'Agrofood_co2_emission.csv' -fires = get_column() +fires = my_utils.get_column(file_name, country_column, country, fires_column) + print(fires)