Merge branch 'main' of https://github.com/alexandersvafdal/Datathon_2024_Borgarlina

data_processing/age_distribution_by_id.py

@@ -1,4 +1,5 @@
 import csv
+import os
 
 def open_file(filename):
     try:
@@ -44,7 +45,7 @@ def get_age_distribution(years, smsv_ids, filename):
 
 if __name__ == '__main__':
     # Example usage:
-    filename = './given_data/ibuafjoldi.csv'
+    filename = os.path.join('given_data', 'ibuafjoldi.csv')
     smsv_ids_to_find = ['0103', '2903', '4002']  # List of desired smsv_ids
     age_dist = get_age_distribution([2017, 2024],smsv_ids_to_find, filename)
 

data_processing/income_decile_by_id.py

@@ -0,0 +1,30 @@
+import csv
+
+def open_file(filename):
+    try:
+        with open(filename, 'r', newline='', encoding='utf-8') as file:
+            csv_reader = csv.reader(file)
+            data = list(csv_reader)
+            return data
+    except FileNotFoundError:
+        print(f"File {filename} not found.")
+        return []
+    except Exception as e:
+        print(f"An error occurred: {e}")
+        return []
+    
+def get_income_decile(years, smsv_ids, filename):
+    income_deciles = {smsv_ids: {year: {} for year in years} for smsv_id in smsv_ids} # Initialize output dict
+    csv_data = open_file(filename)
+
+    if csv_data(): # Check if data is empty
+        header = csv_data[0] # Extract header for column indexing
+        smsv_id_index = header.index('smasvaedi')
+        decile_index = header.index('tekjutiund')
+           
+    
+
+if __name__ == '__main__':
+    # Example usage:
+    filename = './given_data/ibuafjoldi.csv'
+    smsv_ids_to_find = ['0103', '2903', '4002']  # List of desired smsv_ids