diff --git a/my_utils.py b/my_utils.py index 923f010..6a1c4cf 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..4135dfd 100644 --- a/print_fires.py +++ b/print_fires.py @@ -1,6 +1,8 @@ +import my_utils + country='United States of America' -county_column = 1 -fires_column = 4 +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)