danieldux commited on
Commit
7fc2f52
1 Parent(s): 10e83ef

Refactor create_hierarchy_dict function to concisely include additional hierarchy levels

Browse files
Files changed (1) hide show
  1. isco.py +1 -9
isco.py CHANGED
@@ -20,19 +20,11 @@ def create_hierarchy_dict(file: str) -> dict:
20
  with open(file, newline="") as csvfile:
21
  reader = csv.DictReader(csvfile)
22
  for row in reader:
23
- # Extract unit group level code (4 digits)
24
  unit_code = row["unit"].zfill(4)
25
-
26
- # Extract the parent code for the unit group level, which is the minor group level (3 digits)
27
  minor_code = unit_code[0:3]
28
-
29
- # Add the unit code to the hierarchy with its parent code
30
- isco_hierarchy[unit_code] = {minor_code}
31
-
32
- # Additionally, we can add the parent's parent codes at the sub-major group level (2 digits) and major group level (1 digit)
33
  sub_major_code = unit_code[0:2]
34
  major_code = unit_code[0]
35
- isco_hierarchy[unit_code].update({major_code, sub_major_code})
36
 
37
  return isco_hierarchy
38
 
 
20
  with open(file, newline="") as csvfile:
21
  reader = csv.DictReader(csvfile)
22
  for row in reader:
 
23
  unit_code = row["unit"].zfill(4)
 
 
24
  minor_code = unit_code[0:3]
 
 
 
 
 
25
  sub_major_code = unit_code[0:2]
26
  major_code = unit_code[0]
27
+ isco_hierarchy[unit_code] = {minor_code, major_code, sub_major_code}
28
 
29
  return isco_hierarchy
30