scuti/stacksurvey
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added param. to save graphs to file.

Browse Source
This commit is contained in:
scuti
2025-04-19 10:23:42 -07:00
parent 59857f8d36
commit e4ba004fec
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
stackoverflow-survey.ipynb
View File

@@ -59,7 +59,7 @@
" )\n",
" return langs_by_popularity\n",
"\n",
"def visualize_langs(langs, langs2, label1 = \"condition1\", label2 = \"condition2\"):\n",
"def visualize_langs(langs, langs2, label1 = \"condition1\", label2 = \"condition2\", saveto=None):\n",
" DOT_COLOR1 = \"lightblue\"\n",
" DOT_COLOR2 = \"red\"\n",
" BG_COLOR = \"black\" \n",
@@ -88,15 +88,21 @@
" \n",
" plt.grid(axis='x', linestyle='--', alpha=0.75) \n",
" plt.title(\"%s vs %s\" % (label1, label2))\n",
" if saveto is not None:\n",
" plt.savefig(saveto)\n",
" del df, df2\n",
"\n",
"l1 = get_langs( so_df )\n",
"l2 = get_langs( so_df, \"LanguageWantToWorkWith\" )\n",
"visualize_langs(l1,l2, label1=\"have worked with\", label2=\"want to work with\")\n",
"visualize_langs(l1,l2, \n",
" label1=\"have worked with\", label2=\"want to work with\",\n",
" saveto=\"images/used-vs-want2use.png\")\n",
"\n",
"l3 = get_langs( so_df, \"LanguageAdmired\")\n",
"l4 = get_langs( so_df, \"LanguageWantToWorkWith\")\n",
"visualize_langs(l3, l4, label1=\"admired\", label2=\"want to work with\")\n",
"visualize_langs(l3, l4, \n",
" label1=\"admired\", label2=\"want to work with\",\n",
" saveto=\"images/admired-vs-want2use.png\")\n",
" "
]
},
@@ -118,7 +124,7 @@
" result[key] = dict1[key] - dict2[key]\n",
" return result\n",
"\n",
"def visualize_diff(diff_dict, color=\"lightblue\"):\n",
"def visualize_diff(diff_dict, color=\"lightblue\", saveto=None):\n",
" diff_sorted = dict(\n",
" sorted(diff_dict.items(), key=lambda item: item[1], reverse=True)\n",
" )\n",
@@ -145,11 +151,12 @@
" if len(positive_values) < len(df):\n",
" # don't draw the line if every value is greater than 0\n",
" plt.axhline(y=df[KEY].tolist().index(lowest) + offset, color='red', linestyle='--')\n",
" pass\n",
" if saveto is not None:\n",
" plt.savefig(saveto)\n",
" \n",
"motiv_diff = get_difference(l2, l1)\n",
"# print(motiv_diff)\n",
"visualize_diff(motiv_diff)\n",
"visualize_diff(motiv_diff, saveto=\"images/delta.png\")\n",
"\n",
"# determine level of hype\n",
"hype = get_difference(l4, l3)\n",