Stack Overflow Survey Analysis

CoreyMSchafer · CoreyMSchafer · commit eed7fce2b989 · 2019-10-17T00:43:29.000-04:00
diff --git a/Python/StackOverflow-2019/sof-demo.py b/Python/StackOverflow-2019/sof-demo.py
@@ -0,0 +1,30 @@
+import csv
+from collections import defaultdict, Counter
+
+with open('data/survey_results_public.csv') as f:
+    csv_reader = csv.DictReader(f)
+
+    dev_type_info = {}
+
+    for line in csv_reader:
+        dev_types = line['DevType'].split(';')
+
+        for dev_type in dev_types:
+            dev_type_info.setdefault(dev_type, {
+                'total': 0,
+                'language_counter': Counter()
+            })
+
+            languages = line['LanguageWorkedWith'].split(';')
+            dev_type_info[dev_type]['language_counter'].update(languages)
+            dev_type_info[dev_type]['total'] += 1
+
+
+for dev_type, info in dev_type_info.items():
+    print(dev_type)
+
+    for language, value in info['language_counter'].most_common(5):
+        language_pct = (value / info['total']) * 100
+        language_pct = round(language_pct, 2)
+
+        print(f'\t{language}: {language_pct}%')
