Analiza_zarobki_absolwentow.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "\n",
    "major_data = pd.read_csv('graduates-major-data.csv', delimiter = \";\", decimal=\",\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {},
   "outputs": [],
   "source": [
    "task_1_data = major_data.loc[(major_data['P_N'] >= 30) & (major_data['P_POZIOM'] == \"2\")]\n",
    "\n",
    "task_1_data = task_1_data.groupby(['P_KIERUNEK_ID', 'P_KIERUNEK_NAZWA', 'P_NAZWA_UCZELNI'], as_index=False)[\"P_ME_ZAR_P5\"].median()\n",
    "\n",
    "task_1_data = task_1_data[['P_KIERUNEK_ID', \"P_KIERUNEK_NAZWA\", \"P_NAZWA_UCZELNI\", \"P_ME_ZAR_P5\"]]\n",
    "\n",
    "task_1_data = task_1_data.sort_values(by='P_ME_ZAR_P5', ascending=False).head(10)\n",
    "\n",
    "task_1_data[\"KIERUNEK\"] = task_1_data[\"P_KIERUNEK_NAZWA\"].astype(\"str\") + \"(\" + task_1_data[\"P_KIERUNEK_ID\"].astype(\"str\") + \")\""
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['Informatyka(29646)',\n",
       " 'Informatyka(22827)',\n",
       " 'Informatyka(23912)',\n",
       " 'Informatyka(7604)',\n",
       " 'Informatyka(29464)',\n",
       " 'Automatyka i robotyka(27839)',\n",
       " 'Informatyka(8507)',\n",
       " 'Informatyka(25706)',\n",
       " 'Informatyka(27860)',\n",
       " 'Informatyka(28422)']"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "list(task_1_data[\"KIERUNEK\"])"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>P_KIERUNEK_ID</th>\n",
       "      <th>P_KIERUNEK_NAZWA</th>\n",
       "      <th>P_NAZWA_UCZELNI</th>\n",
       "      <th>P_ME_ZAR_P5</th>\n",
       "      <th>KIERUNEK</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>1867</th>\n",
       "      <td>29646</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Akademia Górniczo-Hutnicza im. Stanisława Stas...</td>\n",
       "      <td>13932.48</td>\n",
       "      <td>Informatyka(29646)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>683</th>\n",
       "      <td>22827</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Politechnika Warszawska</td>\n",
       "      <td>12962.67</td>\n",
       "      <td>Informatyka(22827)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>922</th>\n",
       "      <td>23912</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Politechnika Poznańska</td>\n",
       "      <td>12874.23</td>\n",
       "      <td>Informatyka(23912)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>33</th>\n",
       "      <td>7604</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Politechnika Warszawska</td>\n",
       "      <td>12832.00</td>\n",
       "      <td>Informatyka(7604)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1856</th>\n",
       "      <td>29464</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Polsko-Japońska Wyższa Szkoła Technik Komputer...</td>\n",
       "      <td>12740.78</td>\n",
       "      <td>Informatyka(29464)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1622</th>\n",
       "      <td>27839</td>\n",
       "      <td>Automatyka i robotyka</td>\n",
       "      <td>Akademia Górniczo-Hutnicza im. Stanisława Stas...</td>\n",
       "      <td>12716.30</td>\n",
       "      <td>Automatyka i robotyka(27839)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>58</th>\n",
       "      <td>8507</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Politechnika Wrocławska</td>\n",
       "      <td>12416.86</td>\n",
       "      <td>Informatyka(8507)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1238</th>\n",
       "      <td>25706</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Politechnika Gdańska</td>\n",
       "      <td>12239.22</td>\n",
       "      <td>Informatyka(25706)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1628</th>\n",
       "      <td>27860</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Akademia Górniczo-Hutnicza im. Stanisława Stas...</td>\n",
       "      <td>12222.68</td>\n",
       "      <td>Informatyka(27860)</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1723</th>\n",
       "      <td>28422</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>Wojskowa Akademia Techniczna im. Jarosława Dąb...</td>\n",
       "      <td>11963.65</td>\n",
       "      <td>Informatyka(28422)</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "      P_KIERUNEK_ID       P_KIERUNEK_NAZWA  \\\n",
       "1867          29646            Informatyka   \n",
       "683           22827            Informatyka   \n",
       "922           23912            Informatyka   \n",
       "33             7604            Informatyka   \n",
       "1856          29464            Informatyka   \n",
       "1622          27839  Automatyka i robotyka   \n",
       "58             8507            Informatyka   \n",
       "1238          25706            Informatyka   \n",
       "1628          27860            Informatyka   \n",
       "1723          28422            Informatyka   \n",
       "\n",
       "                                        P_NAZWA_UCZELNI  P_ME_ZAR_P5  \\\n",
       "1867  Akademia Górniczo-Hutnicza im. Stanisława Stas...     13932.48   \n",
       "683                             Politechnika Warszawska     12962.67   \n",
       "922                              Politechnika Poznańska     12874.23   \n",
       "33                              Politechnika Warszawska     12832.00   \n",
       "1856  Polsko-Japońska Wyższa Szkoła Technik Komputer...     12740.78   \n",
       "1622  Akademia Górniczo-Hutnicza im. Stanisława Stas...     12716.30   \n",
       "58                              Politechnika Wrocławska     12416.86   \n",
       "1238                               Politechnika Gdańska     12239.22   \n",
       "1628  Akademia Górniczo-Hutnicza im. Stanisława Stas...     12222.68   \n",
       "1723  Wojskowa Akademia Techniczna im. Jarosława Dąb...     11963.65   \n",
       "\n",
       "                          KIERUNEK  \n",
       "1867            Informatyka(29646)  \n",
       "683             Informatyka(22827)  \n",
       "922             Informatyka(23912)  \n",
       "33               Informatyka(7604)  \n",
       "1856            Informatyka(29464)  \n",
       "1622  Automatyka i robotyka(27839)  \n",
       "58               Informatyka(8507)  \n",
       "1238            Informatyka(25706)  \n",
       "1628            Informatyka(27860)  \n",
       "1723            Informatyka(28422)  "
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "task_1_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'TOP_10_Kierunków_zarobki.html'"
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import plotly.express as px\n",
    "import plotly\n",
    "\n",
    "graph_data = task_1_data\n",
    "\n",
    "fig = px.bar(\n",
    "    graph_data, \n",
    "    x = 'KIERUNEK', \n",
    "    y = 'P_ME_ZAR_P5',\n",
    "    color = 'P_NAZWA_UCZELNI',\n",
    "    \n",
    "    category_orders=dict(group=list(task_1_data[\"KIERUNEK\"])),\n",
    "    \n",
    "    title = \"TOP 10 Kierunków z najwyższymi zarobkami (5 lat po ukończeniu studiów II stopnia)\",\n",
    "\n",
    "    labels={\n",
    "        \"P_ME_ZAR_P5\": \"Średnie zarobki po 5 latach od ukończenia studiów [PLN]\",\n",
    "        \"KIERUNEK\": \"Kierunek\",\n",
    "        \"P_NAZWA_UCZELNI\": \"Szkoła wyższa\",\n",
    "    }\n",
    ")\n",
    "\n",
    "fig.update_layout(xaxis_categoryorder = 'total descending')\n",
    "\n",
    "plotly.offline.plot(fig, filename='TOP_10_Kierunków_zarobki.html')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/html": [
       "<div>\n",
       "<style scoped>\n",
       "    .dataframe tbody tr th:only-of-type {\n",
       "        vertical-align: middle;\n",
       "    }\n",
       "\n",
       "    .dataframe tbody tr th {\n",
       "        vertical-align: top;\n",
       "    }\n",
       "\n",
       "    .dataframe thead th {\n",
       "        text-align: right;\n",
       "    }\n",
       "</style>\n",
       "<table border=\"1\" class=\"dataframe\">\n",
       "  <thead>\n",
       "    <tr style=\"text-align: right;\">\n",
       "      <th></th>\n",
       "      <th>P_N</th>\n",
       "      <th>P_POZIOM</th>\n",
       "      <th>P_POZIOM_TEKST_PL</th>\n",
       "      <th>P_N_SEMESTR</th>\n",
       "      <th>P_NAZWA_UCZELNI</th>\n",
       "      <th>P_KIERUNEK_NAZWA</th>\n",
       "      <th>P_WWZ_MIES_60</th>\n",
       "      <th>P_E_ZAR_ETAT_P5</th>\n",
       "      <th>P_ZAR_ETAT_MAX</th>\n",
       "      <th>P_PROC_MIES_BEZR</th>\n",
       "      <th>P_PROC_MIES_BEZR_DOSW</th>\n",
       "      <th>P_WWB</th>\n",
       "      <th>P_ME_ZAR</th>\n",
       "      <th>P_ME_ZAR_P1</th>\n",
       "      <th>P_ME_ZAR_P5</th>\n",
       "      <th>ZWROT_INWESTYCJI_ABSOLUTNY</th>\n",
       "      <th>ZWROT_INWESTYCJI_SZYBKOSC</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>49</th>\n",
       "      <td>84</td>\n",
       "      <td>1</td>\n",
       "      <td>Studia pierwszego stopnia</td>\n",
       "      <td>9</td>\n",
       "      <td>Politechnika Poznańska</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>1.80</td>\n",
       "      <td>8455.20</td>\n",
       "      <td>14783.40</td>\n",
       "      <td>0.5</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.09</td>\n",
       "      <td>5582.42</td>\n",
       "      <td>3939.88</td>\n",
       "      <td>7689.11</td>\n",
       "      <td>620.268889</td>\n",
       "      <td>416.581111</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>52</th>\n",
       "      <td>35</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>3</td>\n",
       "      <td>Politechnika Wrocławska</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>2.67</td>\n",
       "      <td>12559.47</td>\n",
       "      <td>14779.35</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.14</td>\n",
       "      <td>7490.32</td>\n",
       "      <td>5986.47</td>\n",
       "      <td>12416.86</td>\n",
       "      <td>2496.773333</td>\n",
       "      <td>2143.463333</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>65</th>\n",
       "      <td>58</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>4</td>\n",
       "      <td>Uniwersytet Ekonomiczny w Krakowie</td>\n",
       "      <td>Informatyka i ekonometria</td>\n",
       "      <td>2.18</td>\n",
       "      <td>8168.58</td>\n",
       "      <td>19388.61</td>\n",
       "      <td>2.7</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.34</td>\n",
       "      <td>6404.01</td>\n",
       "      <td>3978.92</td>\n",
       "      <td>7751.39</td>\n",
       "      <td>1601.002500</td>\n",
       "      <td>943.117500</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>111</th>\n",
       "      <td>40</td>\n",
       "      <td>1</td>\n",
       "      <td>Studia pierwszego stopnia</td>\n",
       "      <td>6</td>\n",
       "      <td>Wyższa Szkoła Bankowa w Poznaniu</td>\n",
       "      <td>Informatyka i ekonometria</td>\n",
       "      <td>1.82</td>\n",
       "      <td>6083.71</td>\n",
       "      <td>18684.57</td>\n",
       "      <td>2.3</td>\n",
       "      <td>1.0</td>\n",
       "      <td>0.32</td>\n",
       "      <td>4514.63</td>\n",
       "      <td>3348.12</td>\n",
       "      <td>4910.88</td>\n",
       "      <td>752.438333</td>\n",
       "      <td>260.460000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>137</th>\n",
       "      <td>127</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>3</td>\n",
       "      <td>Politechnika Śląska</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>1.84</td>\n",
       "      <td>8969.02</td>\n",
       "      <td>13587.34</td>\n",
       "      <td>0.6</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.11</td>\n",
       "      <td>6157.27</td>\n",
       "      <td>3973.81</td>\n",
       "      <td>8706.79</td>\n",
       "      <td>2052.423333</td>\n",
       "      <td>1577.660000</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>...</th>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "      <td>...</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34699</th>\n",
       "      <td>26</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>3</td>\n",
       "      <td>Państwowa Wyższa Szkoła Informatyki i Przedsię...</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>12501.23</td>\n",
       "      <td>8.4</td>\n",
       "      <td>4.8</td>\n",
       "      <td>0.91</td>\n",
       "      <td>4319.43</td>\n",
       "      <td>3889.23</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1439.810000</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34742</th>\n",
       "      <td>47</td>\n",
       "      <td>1</td>\n",
       "      <td>Studia pierwszego stopnia</td>\n",
       "      <td>7</td>\n",
       "      <td>Politechnika Koszalińska</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>12169.70</td>\n",
       "      <td>3.1</td>\n",
       "      <td>1.7</td>\n",
       "      <td>0.81</td>\n",
       "      <td>3595.09</td>\n",
       "      <td>3160.98</td>\n",
       "      <td>NaN</td>\n",
       "      <td>513.584286</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34813</th>\n",
       "      <td>42</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>4</td>\n",
       "      <td>Politechnika Wrocławska</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>15213.19</td>\n",
       "      <td>0.6</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.10</td>\n",
       "      <td>9113.31</td>\n",
       "      <td>8699.17</td>\n",
       "      <td>NaN</td>\n",
       "      <td>2278.327500</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34815</th>\n",
       "      <td>51</td>\n",
       "      <td>2</td>\n",
       "      <td>Studia drugiego stopnia</td>\n",
       "      <td>4</td>\n",
       "      <td>Uniwersytet Ekonomiczny we Wrocławiu</td>\n",
       "      <td>Informatyka w biznesie - k. unikatowy</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>16062.08</td>\n",
       "      <td>0.4</td>\n",
       "      <td>0.0</td>\n",
       "      <td>0.16</td>\n",
       "      <td>6387.61</td>\n",
       "      <td>6594.70</td>\n",
       "      <td>NaN</td>\n",
       "      <td>1596.902500</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>34831</th>\n",
       "      <td>51</td>\n",
       "      <td>1</td>\n",
       "      <td>Studia pierwszego stopnia</td>\n",
       "      <td>7</td>\n",
       "      <td>Państwowa Wyższa Szkoła Zawodowa im. Stanisław...</td>\n",
       "      <td>Informatyka</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>5907.24</td>\n",
       "      <td>14.8</td>\n",
       "      <td>0.6</td>\n",
       "      <td>2.96</td>\n",
       "      <td>2713.56</td>\n",
       "      <td>2274.96</td>\n",
       "      <td>NaN</td>\n",
       "      <td>387.651429</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>872 rows × 17 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "       P_N P_POZIOM          P_POZIOM_TEKST_PL  P_N_SEMESTR  \\\n",
       "49      84        1  Studia pierwszego stopnia            9   \n",
       "52      35        2    Studia drugiego stopnia            3   \n",
       "65      58        2    Studia drugiego stopnia            4   \n",
       "111     40        1  Studia pierwszego stopnia            6   \n",
       "137    127        2    Studia drugiego stopnia            3   \n",
       "...    ...      ...                        ...          ...   \n",
       "34699   26        2    Studia drugiego stopnia            3   \n",
       "34742   47        1  Studia pierwszego stopnia            7   \n",
       "34813   42        2    Studia drugiego stopnia            4   \n",
       "34815   51        2    Studia drugiego stopnia            4   \n",
       "34831   51        1  Studia pierwszego stopnia            7   \n",
       "\n",
       "                                         P_NAZWA_UCZELNI  \\\n",
       "49                                Politechnika Poznańska   \n",
       "52                               Politechnika Wrocławska   \n",
       "65                    Uniwersytet Ekonomiczny w Krakowie   \n",
       "111                     Wyższa Szkoła Bankowa w Poznaniu   \n",
       "137                                  Politechnika Śląska   \n",
       "...                                                  ...   \n",
       "34699  Państwowa Wyższa Szkoła Informatyki i Przedsię...   \n",
       "34742                           Politechnika Koszalińska   \n",
       "34813                            Politechnika Wrocławska   \n",
       "34815               Uniwersytet Ekonomiczny we Wrocławiu   \n",
       "34831  Państwowa Wyższa Szkoła Zawodowa im. Stanisław...   \n",
       "\n",
       "                            P_KIERUNEK_NAZWA  P_WWZ_MIES_60  P_E_ZAR_ETAT_P5  \\\n",
       "49                               Informatyka           1.80          8455.20   \n",
       "52                               Informatyka           2.67         12559.47   \n",
       "65                 Informatyka i ekonometria           2.18          8168.58   \n",
       "111                Informatyka i ekonometria           1.82          6083.71   \n",
       "137                              Informatyka           1.84          8969.02   \n",
       "...                                      ...            ...              ...   \n",
       "34699                            Informatyka            NaN              NaN   \n",
       "34742                            Informatyka            NaN              NaN   \n",
       "34813                            Informatyka            NaN              NaN   \n",
       "34815  Informatyka w biznesie - k. unikatowy            NaN              NaN   \n",
       "34831                            Informatyka            NaN              NaN   \n",
       "\n",
       "       P_ZAR_ETAT_MAX  P_PROC_MIES_BEZR  P_PROC_MIES_BEZR_DOSW  P_WWB  \\\n",
       "49           14783.40               0.5                    0.0   0.09   \n",
       "52           14779.35               1.0                    0.0   0.14   \n",
       "65           19388.61               2.7                    0.0   0.34   \n",
       "111          18684.57               2.3                    1.0   0.32   \n",
       "137          13587.34               0.6                    0.0   0.11   \n",
       "...               ...               ...                    ...    ...   \n",
       "34699        12501.23               8.4                    4.8   0.91   \n",
       "34742        12169.70               3.1                    1.7   0.81   \n",
       "34813        15213.19               0.6                    0.0   0.10   \n",
       "34815        16062.08               0.4                    0.0   0.16   \n",
       "34831         5907.24              14.8                    0.6   2.96   \n",
       "\n",
       "       P_ME_ZAR  P_ME_ZAR_P1  P_ME_ZAR_P5  ZWROT_INWESTYCJI_ABSOLUTNY  \\\n",
       "49      5582.42      3939.88      7689.11                  620.268889   \n",
       "52      7490.32      5986.47     12416.86                 2496.773333   \n",
       "65      6404.01      3978.92      7751.39                 1601.002500   \n",
       "111     4514.63      3348.12      4910.88                  752.438333   \n",
       "137     6157.27      3973.81      8706.79                 2052.423333   \n",
       "...         ...          ...          ...                         ...   \n",
       "34699   4319.43      3889.23          NaN                 1439.810000   \n",
       "34742   3595.09      3160.98          NaN                  513.584286   \n",
       "34813   9113.31      8699.17          NaN                 2278.327500   \n",
       "34815   6387.61      6594.70          NaN                 1596.902500   \n",
       "34831   2713.56      2274.96          NaN                  387.651429   \n",
       "\n",
       "       ZWROT_INWESTYCJI_SZYBKOSC  \n",
       "49                    416.581111  \n",
       "52                   2143.463333  \n",
       "65                    943.117500  \n",
       "111                   260.460000  \n",
       "137                  1577.660000  \n",
       "...                          ...  \n",
       "34699                        NaN  \n",
       "34742                        NaN  \n",
       "34813                        NaN  \n",
       "34815                        NaN  \n",
       "34831                        NaN  \n",
       "\n",
       "[872 rows x 17 columns]"
      ]
     },
     "execution_count": 6,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "task_2_data = major_data.loc[\n",
    "    (major_data[\"P_KIERUNEK_NAZWA\"].str.lower().str.contains(\"informatyka\"))\n",
    "    & (major_data['P_N'].astype(\"int\") > 25)\n",
    "    & ((major_data['P_POZIOM'] == \"1\") | (major_data['P_POZIOM'] == \"2\"))\n",
    "]\n",
    "\n",
    "task_2_data = task_2_data[[\n",
    "    'P_N', \n",
    "    \n",
    "    'P_POZIOM', \n",
    "    'P_POZIOM_TEKST_PL',\n",
    "    'P_N_SEMESTR',\n",
    "    \n",
    "    'P_NAZWA_UCZELNI',\n",
    "    'P_KIERUNEK_NAZWA',\n",
    "    \n",
    "    'P_WWZ_MIES_60', \n",
    "    'P_E_ZAR_ETAT_P5', \n",
    "    'P_ZAR_ETAT_MAX', \n",
    "    'P_PROC_MIES_BEZR',\n",
    "#     'P_PROC_MIES_BEZR_P1',\n",
    "#     'P_PROC_MIES_BEZR_P2',\n",
    "#     'P_PROC_MIES_BEZR_P3',\n",
    "#     'P_PROC_MIES_BEZR_P4',\n",
    "#     'P_PROC_MIES_BEZR_P5',\n",
    "    'P_PROC_MIES_BEZR_DOSW',\n",
    "#     'P_PROC_MIES_BEZR_DOSW_P1',\n",
    "#     'P_PROC_MIES_BEZR_DOSW_P2',\n",
    "#     'P_PROC_MIES_BEZR_DOSW_P3',\n",
    "#     'P_PROC_MIES_BEZR_DOSW_P4',\n",
    "#     'P_PROC_MIES_BEZR_DOSW_P5',\n",
    "    \n",
    "    'P_WWB',\n",
    "#     'P_WWB_P1',\n",
    "#     'P_WWB_P2',\n",
    "#     'P_WWB_P3',\n",
    "#     'P_WWB_P4',\n",
    "#     'P_WWB_P5',\n",
    "    \n",
    "#     'P_CZY_ETAT',\n",
    "#     'P_CZY_ETAT_P1',\n",
    "#     'P_CZY_ETAT_P2',\n",
    "#     'P_CZY_ETAT_P3',\n",
    "#     'P_CZY_ETAT_P4',\n",
    "#     'P_CZY_ETAT_P5',\n",
    "    \n",
    "    'P_ME_ZAR',\n",
    "    'P_ME_ZAR_P1',\n",
    "    'P_ME_ZAR_P5',\n",
    "   ]]\n",
    "\n",
    "task_2_data[\"ZWROT_INWESTYCJI_ABSOLUTNY\"] = task_2_data['P_ME_ZAR']/task_2_data['P_N_SEMESTR']\n",
    "task_2_data[\"ZWROT_INWESTYCJI_SZYBKOSC\"] = (task_2_data['P_ME_ZAR_P5'] - task_2_data['P_ME_ZAR_P1'])/task_2_data['P_N_SEMESTR']\n",
    "\n",
    "task_2_data"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Średnie_zarobki.html'"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import plotly.express as px\n",
    "import plotly\n",
    "\n",
    "graph_data = task_2_data[['P_ME_ZAR', 'P_KIERUNEK_NAZWA', 'P_NAZWA_UCZELNI', 'P_POZIOM']]\n",
    "\n",
    "fig = px.scatter(\n",
    "    graph_data, \n",
    "    x = 'P_ME_ZAR', \n",
    "    y = 'P_KIERUNEK_NAZWA',\n",
    "    color = 'P_NAZWA_UCZELNI',\n",
    "    facet_col = 'P_POZIOM', \n",
    "        \n",
    "    title = \"Zarobki na studiach informatycznych\",\n",
    "\n",
    "    labels={\n",
    "        \"P_ME_ZAR\": \"Średnie zarobki [PLN]\",\n",
    "        \"P_KIERUNEK_NAZWA\": \"Kierunek\",\n",
    "        \"P_NAZWA_UCZELNI\": \"Szkoła wyższa\",\n",
    "        \"P_POZIOM\": \"Stopień studiów\"\n",
    "    }\n",
    ")\n",
    "\n",
    "plotly.offline.plot(fig, filename='Średnie_zarobki.html')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Zwrot_inwestycji_absolutny.html'"
      ]
     },
     "execution_count": 8,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import plotly.express as px\n",
    "import plotly\n",
    "\n",
    "graph_data = task_2_data[['ZWROT_INWESTYCJI_ABSOLUTNY', 'P_KIERUNEK_NAZWA', 'P_NAZWA_UCZELNI', 'P_POZIOM']]\n",
    "\n",
    "fig = px.scatter(\n",
    "    graph_data, \n",
    "    x = 'ZWROT_INWESTYCJI_ABSOLUTNY', \n",
    "    y = 'P_KIERUNEK_NAZWA',\n",
    "    color = 'P_NAZWA_UCZELNI',\n",
    "    facet_col = 'P_POZIOM', \n",
    "    \n",
    "    title = \"Zwrot inwestycji (absolutny)\",\n",
    "\n",
    "    labels={\n",
    "        \"ZWROT_INWESTYCJI_ABSOLUTNY\": \"Średnie zarobki / liczba semestrów\",\n",
    "        \"P_KIERUNEK_NAZWA\": \"Kierunek\",\n",
    "        \"P_NAZWA_UCZELNI\": \"Szkoła wyższa\",\n",
    "        \"P_POZIOM\": \"Stopień studiów\"\n",
    "    }\n",
    ")\n",
    "\n",
    "plotly.offline.plot(fig, filename='Zwrot_inwestycji_absolutny.html')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 9,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Zwrot_inwestycji_szybkość.html'"
      ]
     },
     "execution_count": 9,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import plotly.express as px\n",
    "import plotly\n",
    "\n",
    "graph_data = task_2_data[['ZWROT_INWESTYCJI_SZYBKOSC', 'P_KIERUNEK_NAZWA', 'P_NAZWA_UCZELNI', 'P_POZIOM']]\n",
    "\n",
    "fig = px.scatter(\n",
    "    graph_data, \n",
    "    x = 'ZWROT_INWESTYCJI_SZYBKOSC', \n",
    "    y = 'P_KIERUNEK_NAZWA',\n",
    "    color = 'P_NAZWA_UCZELNI',\n",
    "    facet_col = 'P_POZIOM', \n",
    "\n",
    "    title = \"Zwrot inwestycji (szybkość)\",\n",
    "    \n",
    "    labels={\n",
    "        \"ZWROT_INWESTYCJI_SZYBKOSC\": \"Wzrost zarobków z 1 do 5 roku po studiach / liczba semestrów\",\n",
    "        \"P_KIERUNEK_NAZWA\": \"Kierunek\",\n",
    "        \"P_NAZWA_UCZELNI\": \"Szkoła wyższa\",\n",
    "        \"P_POZIOM\": \"Stopień studiów\"\n",
    "    }\n",
    ")\n",
    "\n",
    "plotly.offline.plot(fig, filename='Zwrot_inwestycji_szybkość.html')"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 10,
   "metadata": {},
   "outputs": [],
   "source": [
    "task_3_data = major_data.loc[\n",
    "    (major_data[\"P_ROKDYP\"] == 2014)\n",
    "    & (major_data[\"P_POZIOM\"] == \"2\")\n",
    "    & (major_data[\"P_WOJ\"] == 14)\n",
    "]\n",
    "\n",
    "task_3_data = task_3_data[[\n",
    "    'P_DZIEDZINA',\n",
    "    \n",
    "    'P_ME_ZAR_P1',\n",
    "    'P_ME_ZAR_P2',\n",
    "    'P_ME_ZAR_P3',\n",
    "    'P_ME_ZAR_P4',\n",
    "    'P_ME_ZAR_P5']]\n",
    "\n",
    "# task_3_data = task_3_data.set_index('P_DZIEDZINA').T\n",
    "task_3_data\n",
    "\n",
    "tmp = pd.DataFrame(columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "\n",
    "for index, row in task_3_data.iterrows():\n",
    "    row_to_append = pd.DataFrame([[str(row['P_DZIEDZINA']), 1, row['P_ME_ZAR_P1']]], columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "    tmp = tmp.append(row_to_append, ignore_index=True)\n",
    "    row_to_append = pd.DataFrame([[str(row['P_DZIEDZINA']), 2, row['P_ME_ZAR_P2']]], columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "    tmp = tmp.append(row_to_append, ignore_index=True)\n",
    "    row_to_append = pd.DataFrame([[str(row['P_DZIEDZINA']), 3, row['P_ME_ZAR_P3']]], columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "    tmp = tmp.append(row_to_append, ignore_index=True)\n",
    "    row_to_append = pd.DataFrame([[str(row['P_DZIEDZINA']), 4, row['P_ME_ZAR_P4']]], columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "    tmp = tmp.append(row_to_append, ignore_index=True)\n",
    "    row_to_append = pd.DataFrame([[str(row['P_DZIEDZINA']), 5, row['P_ME_ZAR_P5']]], columns=['P_DZIEDZINA', \"PO_ROKU\", 'P_ME_ZAR'])\n",
    "    tmp = tmp.append(row_to_append, ignore_index=True)\n",
    "    \n",
    "task_3_data = tmp[tmp[\"P_DZIEDZINA\"] != \"nan\"]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "metadata": {
    "scrolled": false
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "'Zarobki_dziedzina.html'"
      ]
     },
     "execution_count": 11,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "import plotly.express as px\n",
    "\n",
    "fig = px.line(\n",
    "    task_3_data, \n",
    "    x = 'PO_ROKU',\n",
    "    y = 'P_ME_ZAR',\n",
    "    color = 'P_DZIEDZINA',\n",
    "    \n",
    "    title = \"Zarobki po ukończeniu studiów wedle dziedziny\",\n",
    "    \n",
    "    labels={\n",
    "        \"PO_ROKU\": \"Po n latach od ukończenia\",\n",
    "        \"P_ME_ZAR\": \"Średnie zarobki [PLN]\",\n",
    "        \"P_DZIEDZINA\": \"Dziedzina nauki\",\n",
    "    }\n",
    ")\n",
    "\n",
    "plotly.offline.plot(fig, filename='Zarobki_dziedzina.html')"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.8"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 4
}