903_zahranici.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 10,
   "id": "5dc8b73c-4aa3-47ee-8d7d-6f2afbbca48c",
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import pandas as pd\n",
    "pd.set_option('display.max_columns', 100)\n",
    "pd.set_option('display.max_rows', 500)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 11,
   "id": "5663adf0-aabe-4849-aadb-032b5b5a0b76",
   "metadata": {},
   "outputs": [],
   "source": [
    "import warnings\n",
    "warnings.simplefilter(action='ignore', category=FutureWarning)\n",
    "warnings.simplefilter(action='ignore', category=UserWarning)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 12,
   "id": "03465b58-c782-44ab-955f-1dea0804d65c",
   "metadata": {},
   "outputs": [],
   "source": [
    "df = pd.read_parquet(os.path.join('data','jizdenky.parquet'))"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 13,
   "id": "f443a56d-5855-4ee9-a510-c36ff15713ac",
   "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>oscrapovano</th>\n",
       "      <th>prodejce</th>\n",
       "      <th>odkud</th>\n",
       "      <th>kam</th>\n",
       "      <th>odjezd</th>\n",
       "      <th>predstih</th>\n",
       "      <th>predstih_d</th>\n",
       "      <th>predstih_h</th>\n",
       "      <th>cena</th>\n",
       "      <th>prostredek</th>\n",
       "      <th>volnych_mist</th>\n",
       "      <th>obsazenost</th>\n",
       "      <th>jizdni_doba</th>\n",
       "      <th>vzdalenost</th>\n",
       "      <th>zpozdeni</th>\n",
       "      <th>cena_poznamka</th>\n",
       "      <th>den</th>\n",
       "      <th>prestupy</th>\n",
       "      <th>vlaky</th>\n",
       "      <th>mistenka_zdarma</th>\n",
       "      <th>nahradni_bus</th>\n",
       "      <th>volna_mista_economy</th>\n",
       "      <th>volna_mista_economy_plus</th>\n",
       "      <th>volna_mista_economy_business</th>\n",
       "      <th>volna_mista_premium</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>141671</th>\n",
       "      <td>2024-11-08 07:36:19</td>\n",
       "      <td>LE</td>\n",
       "      <td>Ostrava hl.n.</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-11-12 03:56:00</td>\n",
       "      <td>3 days 20:19:41</td>\n",
       "      <td>3</td>\n",
       "      <td>92.328056</td>\n",
       "      <td>199.0</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>út</td>\n",
       "      <td>-1</td>\n",
       "      <td>[]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>141672</th>\n",
       "      <td>2024-11-08 07:36:19</td>\n",
       "      <td>LE</td>\n",
       "      <td>Ostrava hl.n.</td>\n",
       "      <td>Praha-Libeň</td>\n",
       "      <td>2024-11-12 03:56:00</td>\n",
       "      <td>3 days 20:19:41</td>\n",
       "      <td>3</td>\n",
       "      <td>92.328056</td>\n",
       "      <td>199.0</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>út</td>\n",
       "      <td>-1</td>\n",
       "      <td>[]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>141673</th>\n",
       "      <td>2024-11-08 07:36:19</td>\n",
       "      <td>LE</td>\n",
       "      <td>Ostrava-Svinov</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-11-12 04:03:00</td>\n",
       "      <td>3 days 20:26:41</td>\n",
       "      <td>3</td>\n",
       "      <td>92.444722</td>\n",
       "      <td>199.0</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>út</td>\n",
       "      <td>-1</td>\n",
       "      <td>[]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>141674</th>\n",
       "      <td>2024-11-08 07:36:19</td>\n",
       "      <td>LE</td>\n",
       "      <td>Ostrava hl.n.</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-11-12 10:57:00</td>\n",
       "      <td>4 days 03:20:41</td>\n",
       "      <td>4</td>\n",
       "      <td>99.344722</td>\n",
       "      <td>419.0</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>út</td>\n",
       "      <td>-1</td>\n",
       "      <td>[]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>141675</th>\n",
       "      <td>2024-11-08 07:36:19</td>\n",
       "      <td>LE</td>\n",
       "      <td>Ostrava hl.n.</td>\n",
       "      <td>Praha-Libeň</td>\n",
       "      <td>2024-11-12 10:57:00</td>\n",
       "      <td>4 days 03:20:41</td>\n",
       "      <td>4</td>\n",
       "      <td>99.344722</td>\n",
       "      <td>419.0</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>út</td>\n",
       "      <td>-1</td>\n",
       "      <td>[]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</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",
       "      <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>998402</th>\n",
       "      <td>2024-11-30 23:25:28</td>\n",
       "      <td>ARR</td>\n",
       "      <td>České Budějovice</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-12-14 20:35:00</td>\n",
       "      <td>13 days 21:09:32</td>\n",
       "      <td>13</td>\n",
       "      <td>333.158889</td>\n",
       "      <td>200.0</td>\n",
       "      <td>vlak</td>\n",
       "      <td>38.0</td>\n",
       "      <td>None</td>\n",
       "      <td>203.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>so</td>\n",
       "      <td>0</td>\n",
       "      <td>[R 1173]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>998403</th>\n",
       "      <td>2024-11-30 23:25:28</td>\n",
       "      <td>ARR</td>\n",
       "      <td>České Budějovice</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-12-14 18:43:00</td>\n",
       "      <td>13 days 19:17:32</td>\n",
       "      <td>13</td>\n",
       "      <td>331.292222</td>\n",
       "      <td>200.0</td>\n",
       "      <td>vlak</td>\n",
       "      <td>33.0</td>\n",
       "      <td>None</td>\n",
       "      <td>211.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>so</td>\n",
       "      <td>0</td>\n",
       "      <td>[R 1171]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>998404</th>\n",
       "      <td>2024-11-30 23:25:28</td>\n",
       "      <td>ARR</td>\n",
       "      <td>České Budějovice</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-12-14 16:43:00</td>\n",
       "      <td>13 days 17:17:32</td>\n",
       "      <td>13</td>\n",
       "      <td>329.292222</td>\n",
       "      <td>200.0</td>\n",
       "      <td>vlak</td>\n",
       "      <td>39.0</td>\n",
       "      <td>None</td>\n",
       "      <td>211.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>so</td>\n",
       "      <td>0</td>\n",
       "      <td>[R 1169]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>998405</th>\n",
       "      <td>2024-11-30 23:25:28</td>\n",
       "      <td>ARR</td>\n",
       "      <td>České Budějovice</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-12-14 12:33:00</td>\n",
       "      <td>13 days 13:07:32</td>\n",
       "      <td>13</td>\n",
       "      <td>325.125556</td>\n",
       "      <td>200.0</td>\n",
       "      <td>vlak</td>\n",
       "      <td>39.0</td>\n",
       "      <td>None</td>\n",
       "      <td>201.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>so</td>\n",
       "      <td>0</td>\n",
       "      <td>[R 1167]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>998406</th>\n",
       "      <td>2024-11-30 23:25:28</td>\n",
       "      <td>ARR</td>\n",
       "      <td>České Budějovice</td>\n",
       "      <td>Praha hl.n.</td>\n",
       "      <td>2024-12-13 20:35:00</td>\n",
       "      <td>12 days 21:09:32</td>\n",
       "      <td>12</td>\n",
       "      <td>309.158889</td>\n",
       "      <td>200.0</td>\n",
       "      <td>vlak</td>\n",
       "      <td>39.0</td>\n",
       "      <td>None</td>\n",
       "      <td>203.0</td>\n",
       "      <td>187.0</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>pá</td>\n",
       "      <td>0</td>\n",
       "      <td>[R 1173]</td>\n",
       "      <td>None</td>\n",
       "      <td>None</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>714958 rows × 25 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "               oscrapovano prodejce             odkud          kam  \\\n",
       "141671 2024-11-08 07:36:19       LE     Ostrava hl.n.  Praha hl.n.   \n",
       "141672 2024-11-08 07:36:19       LE     Ostrava hl.n.  Praha-Libeň   \n",
       "141673 2024-11-08 07:36:19       LE    Ostrava-Svinov  Praha hl.n.   \n",
       "141674 2024-11-08 07:36:19       LE     Ostrava hl.n.  Praha hl.n.   \n",
       "141675 2024-11-08 07:36:19       LE     Ostrava hl.n.  Praha-Libeň   \n",
       "...                    ...      ...               ...          ...   \n",
       "998402 2024-11-30 23:25:28      ARR  České Budějovice  Praha hl.n.   \n",
       "998403 2024-11-30 23:25:28      ARR  České Budějovice  Praha hl.n.   \n",
       "998404 2024-11-30 23:25:28      ARR  České Budějovice  Praha hl.n.   \n",
       "998405 2024-11-30 23:25:28      ARR  České Budějovice  Praha hl.n.   \n",
       "998406 2024-11-30 23:25:28      ARR  České Budějovice  Praha hl.n.   \n",
       "\n",
       "                    odjezd         predstih  predstih_d  predstih_h   cena  \\\n",
       "141671 2024-11-12 03:56:00  3 days 20:19:41           3   92.328056  199.0   \n",
       "141672 2024-11-12 03:56:00  3 days 20:19:41           3   92.328056  199.0   \n",
       "141673 2024-11-12 04:03:00  3 days 20:26:41           3   92.444722  199.0   \n",
       "141674 2024-11-12 10:57:00  4 days 03:20:41           4   99.344722  419.0   \n",
       "141675 2024-11-12 10:57:00  4 days 03:20:41           4   99.344722  419.0   \n",
       "...                    ...              ...         ...         ...    ...   \n",
       "998402 2024-12-14 20:35:00 13 days 21:09:32          13  333.158889  200.0   \n",
       "998403 2024-12-14 18:43:00 13 days 19:17:32          13  331.292222  200.0   \n",
       "998404 2024-12-14 16:43:00 13 days 17:17:32          13  329.292222  200.0   \n",
       "998405 2024-12-14 12:33:00 13 days 13:07:32          13  325.125556  200.0   \n",
       "998406 2024-12-13 20:35:00 12 days 21:09:32          12  309.158889  200.0   \n",
       "\n",
       "       prostredek  volnych_mist obsazenost  jizdni_doba  vzdalenost zpozdeni  \\\n",
       "141671       None           NaN       None          NaN         NaN     None   \n",
       "141672       None           NaN       None          NaN         NaN     None   \n",
       "141673       None           NaN       None          NaN         NaN     None   \n",
       "141674       None           NaN       None          NaN         NaN     None   \n",
       "141675       None           NaN       None          NaN         NaN     None   \n",
       "...           ...           ...        ...          ...         ...      ...   \n",
       "998402       vlak          38.0       None        203.0       187.0     None   \n",
       "998403       vlak          33.0       None        211.0       187.0     None   \n",
       "998404       vlak          39.0       None        211.0       187.0     None   \n",
       "998405       vlak          39.0       None        201.0       187.0     None   \n",
       "998406       vlak          39.0       None        203.0       187.0     None   \n",
       "\n",
       "       cena_poznamka den  prestupy     vlaky mistenka_zdarma nahradni_bus  \\\n",
       "141671          None  út        -1        []            None         None   \n",
       "141672          None  út        -1        []            None         None   \n",
       "141673          None  út        -1        []            None         None   \n",
       "141674          None  út        -1        []            None         None   \n",
       "141675          None  út        -1        []            None         None   \n",
       "...              ...  ..       ...       ...             ...          ...   \n",
       "998402          None  so         0  [R 1173]            None         None   \n",
       "998403          None  so         0  [R 1171]            None         None   \n",
       "998404          None  so         0  [R 1169]            None         None   \n",
       "998405          None  so         0  [R 1167]            None         None   \n",
       "998406          None  pá         0  [R 1173]            None         None   \n",
       "\n",
       "        volna_mista_economy  volna_mista_economy_plus  \\\n",
       "141671                  NaN                       NaN   \n",
       "141672                  NaN                       NaN   \n",
       "141673                  NaN                       NaN   \n",
       "141674                  NaN                       NaN   \n",
       "141675                  NaN                       NaN   \n",
       "...                     ...                       ...   \n",
       "998402                  NaN                       NaN   \n",
       "998403                  NaN                       NaN   \n",
       "998404                  NaN                       NaN   \n",
       "998405                  NaN                       NaN   \n",
       "998406                  NaN                       NaN   \n",
       "\n",
       "        volna_mista_economy_business  volna_mista_premium  \n",
       "141671                           NaN                  NaN  \n",
       "141672                           NaN                  NaN  \n",
       "141673                           NaN                  NaN  \n",
       "141674                           NaN                  NaN  \n",
       "141675                           NaN                  NaN  \n",
       "...                              ...                  ...  \n",
       "998402                           NaN                  NaN  \n",
       "998403                           NaN                  NaN  \n",
       "998404                           NaN                  NaN  \n",
       "998405                           NaN                  NaN  \n",
       "998406                           NaN                  NaN  \n",
       "\n",
       "[714958 rows x 25 columns]"
      ]
     },
     "execution_count": 13,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df[df['oscrapovano'] <= '2024-12-01']"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 14,
   "id": "7763f1da-f649-4a38-9d33-1942c3905a72",
   "metadata": {},
   "outputs": [],
   "source": [
    "df = df[df['predstih_h'] >= 0]"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "8d5de84f-de38-4791-9033-c0a95c69a403",
   "metadata": {},
   "source": [
    "Následující funkce vyfiltruje spojení mezi body A a B a nazpátek nabízené dopravcem C. Parametr \"nasobek\" určuje, o kolik delší může jízda být oproti nejkratší nalezené."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 16,
   "id": "4ae26759-e1c6-413f-8bd1-1a1e7839c684",
   "metadata": {},
   "outputs": [],
   "source": [
    "def filtr(dopravce, mesto1, mesto2, nasobek=1.25):\n",
    "    dfc = df.copy()\n",
    "    print(f\"{dopravce}: {mesto1}-{mesto2}\")\n",
    "    dfc = dfc[dfc['prodejce'] == dopravce]\n",
    "    dfc = dfc[dfc['odkud'].str.contains(mesto1) | dfc['kam'].str.contains(mesto1)]\n",
    "    dfc = dfc[dfc['odkud'].str.contains(mesto2) | dfc['kam'].str.contains(mesto2)]\n",
    "    minimum_prestupu = dfc['prestupy'].min()\n",
    "    if minimum_prestupu == -1:\n",
    "        print(\"Pozor, vylezlo nám tu -1 přestupů. Opravuji.\")\n",
    "        minimum_prestupu = 0\n",
    "    print(f\"Nejmenší počet přestupů: {minimum_prestupu}.\")\n",
    "    minimum_casu = dfc['jizdni_doba'].min()\n",
    "    print(f\"Nejrychlejší jízdní doba: {minimum_casu} min.\")\n",
    "    dfc = dfc[(dfc['prestupy'] == minimum_prestupu) & (df['jizdni_doba'] <= (minimum_casu * nasobek))]\n",
    "    dfc = dfc.drop_duplicates(subset=['odjezd','kam','predstih_d'], keep='last')\n",
    "    print(f\"Celkem řádků: {len(dfc)}\\n\")\n",
    "    return dfc.reset_index(drop=True)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 17,
   "id": "28726b7e-d00d-4670-be2c-7854b3b7f16b",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "prodejce  odkud                             kam                          \n",
       "ČD        Praha hl.n.                       Brno hl.n.                       66588\n",
       "          Brno hl.n.                        Praha hl.n.                      64827\n",
       "          Praha hl.n.                       Ostrava hl.n.                    52504\n",
       "          Ostrava hl.n.                     Praha hl.n.                      49478\n",
       "RJ        Brno                              Praha                            29357\n",
       "          Praha                             Brno                             27064\n",
       "          Ostrava                           Praha                            18427\n",
       "ARR       Pardubice hl.n.                   Liberec                          16663\n",
       "          Liberec                           Pardubice hl.n.                  16230\n",
       "RJ        Praha                             Ostrava                          15327\n",
       "ČD        Praha hl.n.                       Bratislava hl.st.                14926\n",
       "          Bratislava hl.st.                 Praha hl.n.                      14370\n",
       "          Praha hl.n.                       Wien Hbf                         13116\n",
       "          Wien Hbf                          Praha hl.n.                      12373\n",
       "          Praha hl.n.                       Warszawa Centralna               12179\n",
       "          Warszawa Centralna                Praha hl.n.                      10963\n",
       "LE        Praha hl.n.                       Ostrava hl.n.                    10795\n",
       "                                            Ostrava-Svinov                   10795\n",
       "          Ostrava hl.n.                     Praha hl.n.                      10545\n",
       "          Ostrava-Svinov                    Praha hl.n.                      10545\n",
       "ČD        Praha hl.n.                       Bruxelles-Midi                   10235\n",
       "ARR       Liberec                           Ústí nad Labem hlavní nádraží     9367\n",
       "ČD        Praha hl.n.                       Budapest-Nyugati pu               9318\n",
       "ARR       Ústí nad Labem hlavní nádraží     Liberec                           8942\n",
       "          Praha hl.n.                       České Budějovice                  8515\n",
       "ČD        Budapest-Nyugati pu               Praha hl.n.                       8388\n",
       "ARR       České Budějovice                  Praha hl.n.                       8304\n",
       "          Tanvald                           Praha hl.n.                       8181\n",
       "ČD        Praha hl.n.                       Budapest-Kelenföld                7854\n",
       "          Bruxelles-Midi                    Praha hl.n.                       7598\n",
       "RJ        Brno                              Vídeň                             7431\n",
       "          Vídeň                             Brno                              6809\n",
       "ČD        Budapest-Kelenföld                Praha hl.n.                       6508\n",
       "ARR       Praha hl.n.                       Tanvald                           6189\n",
       "RJ        Brno                              Ostrava                           6042\n",
       "          Ostrava                           Brno                              5966\n",
       "ČD        Ústí n.L.hl.n.                    Dresden Hbf                       4876\n",
       "          Dresden Hbf                       Ústí n.L.hl.n.                    4717\n",
       "          Pardubice hl.n.                   Košice                            4295\n",
       "          Plzeň hl.n.                       Slavonice                         3907\n",
       "          Praha hl.n.                       Krakow Glowny                     3546\n",
       "          Slavonice                         Plzeň hl.n.                       3521\n",
       "RJ        Praha                             Budapešť                          3487\n",
       "ČD        Praha hl.n.                       Ljubljana                         3210\n",
       "          Krakow Glowny                     Praha hl.n.                       3169\n",
       "          Ljubljana                         Praha hl.n.                       3064\n",
       "RJ        Praha                             Bratislava                        2962\n",
       "ČD        Ostrava hl.n.                     Karlovy Vary                      2515\n",
       "          Berlin Hbf                        Warszawa Centralna                2281\n",
       "          Brno hl.n.                        Ljubljana                         2260\n",
       "          Berlin Hbf                        Bruxelles-Midi                    2246\n",
       "RJ        Praha                             Vídeň                             2242\n",
       "ČD        Košice                            Pardubice hl.n.                   2173\n",
       "          Bern                              Praha hl.n.                       2170\n",
       "          Karlovy Vary                      Ostrava hl.n.                     2135\n",
       "          Praha-Holešovice                  Ostrava hl.n.                     2126\n",
       "          Praha hl.n.                       Bern                              2051\n",
       "          Köln Messe/Deutz                  Bruxelles-Midi                    2034\n",
       "          Praha hl.n.                       Bratislava-Petržalka              1989\n",
       "          Hamburg Hbf                       Brno hl.n.                        1971\n",
       "          Praha hl.n.                       Wien Franz-Josefs-Bahnhof         1953\n",
       "          Ljubljana                         Brno hl.n.                        1919\n",
       "          Wien Franz-Josefs-Bahnhof         Praha hl.n.                       1905\n",
       "RJ        Vídeň                             Praha                             1874\n",
       "ČD        Olomouc hl.n.                     Przemysl Gl.                      1859\n",
       "          Brno hl.n.                        Hamburg Hbf                       1844\n",
       "LE        Pardubice hl.n.                   Košice                            1803\n",
       "ČD        Praha hl.n.                       Ostrava-Svinov                    1780\n",
       "                                            Berlin Hbf                        1775\n",
       "          Ostrava hl.n.                     Praha-Holešovice                  1769\n",
       "LE        Košice                            Pardubice hl.n.                   1750\n",
       "ČD        Ostrava-Svinov                    Praha hl.n.                       1747\n",
       "          Praha-Holešovice                  Brno hl.n.                        1636\n",
       "RJ        Bratislava                        Praha                             1629\n",
       "          Praha                             Košice                            1553\n",
       "ČD        Brno hl.n.                        Praha-Holešovice                  1552\n",
       "          Ostrava hl.n.                     Gdansk Glowny                     1502\n",
       "          Berlin Hbf                        Praha hl.n.                       1496\n",
       "          Köln Messe/Deutz Gl.11-12         Praha hl.n.                       1456\n",
       "          České Budějovice                  Zürich HB                         1314\n",
       "          Budapest-Deli pu                  Praha hl.n.                       1257\n",
       "          Praha-Holešovice                  Bratislava hl.st.                 1182\n",
       "          Gdansk Glowny                     Ostrava hl.n.                     1163\n",
       "          Bratislava-Petržalka              Praha hl.n.                       1155\n",
       "          Przemysl Gl.                      Olomouc hl.n.                     1128\n",
       "          Plzeň hl.n.                       Ostrava hl.n.                     1059\n",
       "          Praha hl.n.                       Berlin Südkreuz                   1022\n",
       "          Bratislava hl.st.                 Praha-Holešovice                  1013\n",
       "          Bruxelles-Midi                    Praha-Holešovice                  1008\n",
       "          Praha hl.n.                       Zagreb Glavni kolodvor             982\n",
       "          Amsterdam Centraal                Praha hl.n.                        960\n",
       "          Praha-Holešovice                  Budapest-Nyugati pu                896\n",
       "          Praha hl.n.                       Wroclaw Glowny                     892\n",
       "          Wroclaw Glowny                    Praha hl.n.                        845\n",
       "          Praha hl.n.                       Amsterdam Centraal                 842\n",
       "          Praha-Holešovice                  Bruxelles-Midi                     824\n",
       "          Plzeň, , Terminál Hlavní nádraží  Karlovy Vary, , terminál           790\n",
       "          Berlin Hbf                        Amsterdam Centraal                 788\n",
       "RJ        Praha                             Krakov                             787\n",
       "          Budapešť                          Praha                              785\n",
       "dtype: int64"
      ]
     },
     "execution_count": 17,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "df.groupby([\"prodejce\",\"odkud\",\"kam\"]).size().nlargest(100)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 18,
   "id": "86606670-d9cc-4fd3-aa6b-228d6b166425",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "ČD: Praha hl.n.-Berlin Hbf\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 252.0 min.\n",
      "Celkem řádků: 1130\n",
      "\n",
      "ČD: Praha hl.n.-Warszawa Centralna\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 479.0 min.\n",
      "Celkem řádků: 3375\n",
      "\n",
      "ČD: Praha hl.n.-Wien Hbf\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 241.0 min.\n",
      "Celkem řádků: 8243\n",
      "\n",
      "RJ: Praha-Víde\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 238.0 min.\n",
      "Celkem řádků: 2086\n",
      "\n",
      "ČD: Praha hl.n.-Bratislava hl.st.\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 253.0 min.\n",
      "Celkem řádků: 11300\n",
      "\n",
      "RJ: Praha-Bratislava\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 250.0 min.\n",
      "Celkem řádků: 2469\n",
      "\n",
      "RJ: Praha-Čop\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 645.0 min.\n",
      "Celkem řádků: 384\n",
      "\n",
      "RJ: Praha-Budap\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 399.0 min.\n",
      "Celkem řádků: 2368\n",
      "\n",
      "ČD: Praha-Budap\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 401.0 min.\n",
      "Celkem řádků: 8464\n",
      "\n",
      "ČD: Ústí-Dresd\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 40.0 min.\n",
      "Celkem řádků: 61\n",
      "\n",
      "LE: Praha-Krak\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 344.0 min.\n",
      "Celkem řádků: 376\n",
      "\n",
      "LE: Pardub-Košic\n",
      "Pozor, vylezlo nám tu -1 přestupů. Opravuji.\n",
      "Nejmenší počet přestupů: 0.\n",
      "Nejrychlejší jízdní doba: 428.0 min.\n",
      "Celkem řádků: 1644\n",
      "\n"
     ]
    }
   ],
   "source": [
    "cd_berlin = filtr(\"ČD\",\"Praha hl.n.\",\"Berlin Hbf\")\n",
    "cd_berlin = pd.Series(cd_berlin.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Praha-Berlín\").head(60)\n",
    "cd_vars = filtr(\"ČD\",\"Praha hl.n.\",\"Warszawa Centralna\")\n",
    "cd_vars = pd.Series(cd_vars.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Praha-Varšava\").head(59)\n",
    "cd_vid = filtr(\"ČD\",\"Praha hl.n.\",\"Wien Hbf\")\n",
    "cd_vid = pd.Series(cd_vid.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Praha-Vídeň\").head(60)\n",
    "rj_vid = filtr(\"RJ\",\"Praha\",\"Víde\")\n",
    "rj_vid = pd.Series(rj_vid.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"RJ Praha-Vídeň\").head(60)\n",
    "cd_brat = filtr(\"ČD\",\"Praha hl.n.\",\"Bratislava hl.st.\")\n",
    "cd_brat = pd.Series(cd_brat.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Praha-Bratislava\").head(60)\n",
    "rj_brat = filtr(\"RJ\",\"Praha\",\"Bratislava\")\n",
    "rj_brat = pd.Series(rj_brat.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"RJ Praha-Bratislava\").head(60)\n",
    "rj_cop = filtr('RJ','Praha','Čop')\n",
    "rj_cop = pd.Series(rj_cop.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"RJ Praha-Čop\").head(60)\n",
    "rj_buda = filtr('RJ','Praha','Budap')\n",
    "rj_buda = pd.Series(rj_buda.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"RJ Praha-Budapešť\").head(60)\n",
    "cd_buda = filtr('ČD','Praha','Budap')\n",
    "cd_buda = pd.Series(cd_buda.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Praha-Budapešť\").head(60)\n",
    "cd_dres = filtr('ČD','Ústí','Dresd')\n",
    "cd_dres = pd.Series(cd_dres.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"ČD Ústí nad Labem-Drážďany\").head(60)\n",
    "le_krak = filtr(\"LE\",\"Praha\",\"Krak\")\n",
    "le_krak = pd.Series(le_krak.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"LE Praha-Krakov\").head(60)\n",
    "le_kosi = filtr(\"LE\",\"Pardub\",\"Košic\")\n",
    "le_kosi = pd.Series(le_kosi.groupby('predstih_d')['cena'].agg(lambda x: [x.max(), x.min(), x.median()]), name=\"LE Pardubice-Košice\").head(60)\n",
    "ukazat = [cd_vars, cd_berlin, cd_brat, rj_brat, cd_vid, rj_vid, le_kosi, rj_cop, cd_buda, rj_buda, cd_dres, le_krak]\n",
    "len(ukazat)\n",
    "for u in ukazat:\n",
    "    u.name = u.name.replace(\"-\",\" &#8596; \")\n",
    "    u.index = u.index.map(lambda x: f\"{x} d\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 19,
   "id": "2d478d5d-fccb-4108-a0d7-6cc6b085c496",
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "['#173a79',\n",
       " '#173a79',\n",
       " '#173a79',\n",
       " '#fabb00',\n",
       " '#173a79',\n",
       " '#fabb00',\n",
       " '#e68014',\n",
       " '#fabb00',\n",
       " '#173a79',\n",
       " '#fabb00',\n",
       " '#173a79',\n",
       " '#e68014']"
      ]
     },
     "execution_count": 19,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "farby = []\n",
    "for u in ukazat:\n",
    "    if \"ČD\" in u.name:\n",
    "        farby.append(\"#173a79\")\n",
    "    elif \"RJ\" in u.name:\n",
    "        farby.append(\"#fabb00\")\n",
    "    elif \"LE\" in u.name:\n",
    "        farby.append(\"#e68014\")\n",
    "farby"
   ]
  },
  {
   "cell_type": "markdown",
   "id": "367ebb02-8dd9-45f5-bac0-6d914e35beaf",
   "metadata": {},
   "source": [
    "Tohle bude úplně odpudivá ad hoc úprava starší funkce pro generování grafů."
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 21,
   "id": "f84084bd-a54a-4de8-99d5-7baf1bcf4ffc",
   "metadata": {},
   "outputs": [],
   "source": [
    "def irozhlas_graf(\n",
    "    carovy=[],\n",
    "    sloupcovy=[],\n",
    "    vodorovny=[],\n",
    "    rozpeti=[],\n",
    "    procenta=[],\n",
    "    skryte=[],\n",
    "    barvy=[\n",
    "            \"#b2e061\",  ## světle zelená (light green)\n",
    "            \"#7eb0d5\",  ## světle modrá (light blue)\n",
    "            \"#fd7f6f\",  ## světle červená (light red)\n",
    "            \"#bd7ebe\",  ## světle fialová (light purple)\n",
    "            \"#ffb55a\",  ## oranžová (orange)\n",
    "            \"#ffee65\",  ## žlutá (yellow)\n",
    "            \"#beb9db\",  ## levandulová (lavender)\n",
    "            \"#fdcce5\",  ## skoro černá\n",
    "            \"#8bd3c7\",  ## světle tyrkysová (light turquoise),\n",
    "        \"red\",\n",
    "        \"blue\",\n",
    "        \"purple\"\n",
    "        ],\n",
    "    histogram=False,\n",
    "    max_procenta=100,\n",
    "    target=\"\",\n",
    "    titulek=\"\",\n",
    "    podtitulek=\"\",\n",
    "    naproti=[],\n",
    "    osay=\" \",\n",
    "    osay2=\" \",\n",
    "    osaymin=None,\n",
    "    osaymax=None,\n",
    "    kredity=[\"zdroj dat a autorstvo\", \"url odkazu\"],\n",
    "    zaokrouhleni=1,\n",
    "    prvni=True,\n",
    "    skladany=False,\n",
    "    naopak=False,\n",
    "    vzhurunohama=False,\n",
    "    skrytnuly=False,\n",
    "):\n",
    "    \"\"\"\n",
    "    Funkce vygeneruje HighCharts graf z pandas Series (jedné nebo více).\n",
    "\n",
    "    iROZHLAS-friendly barvy:\n",
    "    - \"#b2e061\" světle zelená\n",
    "    - \"#7eb0d5\" světle modrá\n",
    "    - \"#fd7f6f\" světle červená\n",
    "    - \"#bd7ebe\" světle fialová\n",
    "    - \"#ffb55a\" oranžová\n",
    "    - \"#ffee65\" žlutá\n",
    "    - \"#beb9db\" levandulová\n",
    "    - \"#fdcce5\" skoro černá\n",
    "    - \"#8bd3c7\" světle tyrkysová\n",
    "    \"\"\"\n",
    "\n",
    "    import os\n",
    "    import pandas as pd\n",
    "    from highcharts_core.chart import Chart\n",
    "    from highcharts_core.options.series.area import LineSeries\n",
    "    from highcharts_core.options.series.bar import ColumnSeries\n",
    "    from highcharts_core.options.series.bar import BarSeries\n",
    "    from highcharts_core.options.series.area import AreaRangeSeries\n",
    "    from highcharts_core.options.series.histogram import HistogramSeries\n",
    "    from highcharts_core.options.legend import Legend\n",
    "    from highcharts_core.options.title import Title\n",
    "    from highcharts_core.options.subtitle import Subtitle\n",
    "    from highcharts_core.options.credits import Credits\n",
    "\n",
    "    pocitadlo_barev = 0\n",
    "    \n",
    "    nastaveni = {}\n",
    "\n",
    "    if prvni:\n",
    "        zdrojaky = f\"\"\"<script src=\"https://code.highcharts.com/highcharts.js\"></script><script src=\"https://code.highcharts.com/highcharts-more.js\"></script><script src=\"https://code.highcharts.com/modules/exporting.js\"></script><script src=\"https://code.highcharts.com/modules/export-data.js\"></script><script src=\"https://code.highcharts.com/modules/accessibility.js\"></script>\n",
    "        <link\n",
    "  rel=\"stylesheet\"\n",
    "  href=\"https://fonts.googleapis.com/css?family=Noticia+Text\"\n",
    "/>\n",
    "<link\n",
    "  href=\"https://fonts.googleapis.com/css2?family=Asap\"\n",
    "  rel=\"stylesheet\"\n",
    "/>\n",
    "<link\n",
    "  href=\"https://fonts.googleapis.com/css2?family=Roboto\"\n",
    "  rel=\"stylesheet\"\n",
    "/>\n",
    "<link\n",
    "  rel=\"stylesheet\"\n",
    "  href=\"https://data.irozhlas.cz/hhighcharts-template/style.css\"\n",
    "/>\n",
    "<link\n",
    "  rel=\"stylesheet\"\n",
    "  href=\"https://data.irozhlas.cz/hhighcharts-template/highcharts-style.css\"\n",
    "/>\n",
    "<style type=\"text/css\">text{{font-family:\"Asap\"!important}}.paragraph{{font-family:\"Noticia text\"!important}}.href{{color:#666;fill:#666}}.highcharts-title{{font-family:\"Noticia text\"!important;font-weight:700!important;text-align:left!important;left:10px!important}}.highcharts-subtitle{{text-align:left!important;font-size:.95rem!important;left:10px!important;font-family:\"Asap\"!important}}.highcharts-data-labels text{{font-size:.85rem!important}}.highcharts-axis-labels text{{font-size:.85rem!important}}text.highcharts-plot-line-label{{font-size:.85rem!important;fill:#666}}text.highcharts-plot-band-label{{font-size:.85rem!important;fill:#666}}text.highcharts-credits{{font-size:.75rem!important}}.highcharts-tooltip span{{font-family:\"Asap\"!important}}.axis-label-on-tick{{fill:#aaa;color:#aaa}}.mock-empty-line{{fill:#fff;color:#fff}}</style>\"\"\"\n",
    "    else:\n",
    "        zdrojaky = \"\"\n",
    "\n",
    "    pred = f\"\"\"{zdrojaky}\n",
    "        <figure id=\"{target}\">\n",
    "        <div id=\"container\"></div>\n",
    "        </figure>\n",
    "        <script>\"\"\"\n",
    "\n",
    "    if len(carovy) > 0:\n",
    "        categories = carovy[0].index.to_list()\n",
    "    if len(sloupcovy) > 0:\n",
    "        categories = sloupcovy[0].index.to_list()\n",
    "    if len(vodorovny) > 0:\n",
    "        categories = vodorovny[0].index.to_list()\n",
    "    if len(rozpeti) > 0:\n",
    "        categories = rozpeti[0].index.to_list()\n",
    "\n",
    "    categories = [str(x) for x in categories]\n",
    "\n",
    "    nastaveni[\"xAxis\"] = {\"categories\": categories, \"min\": 0}\n",
    "    nastaveni[\"yAxis\"] = [\n",
    "        {\n",
    "            \"title\": {\"text\": osay},\n",
    "            \"reversed\": vzhurunohama,\n",
    "            \"max\": osaymax,\n",
    "            \"min\": osaymin,\n",
    "        }\n",
    "    ]\n",
    "\n",
    "    if skladany:\n",
    "        if len(sloupcovy) > 0:\n",
    "            nastaveni[\"plotOptions\"] = {\"column\": {\"stacking\": \"normal\"}}\n",
    "        if len(vodorovny) > 0:\n",
    "            nastaveni[\"plotOptions\"] = {\"bar\": {\"stacking\": \"normal\"}}\n",
    "    if histogram:\n",
    "        nastaveni[\"plotOptions\"] = {\n",
    "            \"column\": {\n",
    "                \"pointPadding\": 0,\n",
    "                \"borderWidth\": 0,\n",
    "                \"groupPadding\": 0,\n",
    "                \"shadow\": False,\n",
    "            }\n",
    "        }\n",
    "\n",
    "    \n",
    "    if len(procenta) > 0:\n",
    "        osa_procent = {\n",
    "            \"title\": {\"text\": osay2},\n",
    "            \"max\": max_procenta,\n",
    "            \"min\": 0,\n",
    "            \"labels\": {\"format\": \"{value} %\"},\n",
    "        }\n",
    "\n",
    "        if len(procenta) != len(carovy) + len(sloupcovy):\n",
    "            osa_procent[\"opposite\"] = True\n",
    "            druha_osa = 1\n",
    "            nastaveni[\"yAxis\"].append(osa_procent)\n",
    "            nastaveni[\"alignTicks\"] = False\n",
    "        if len(procenta) == len(carovy) + len(sloupcovy):\n",
    "            nastaveni[\"yAxis\"] = [osa_procent]\n",
    "            druha_osa = 0\n",
    "\n",
    "    if len(naproti) > 0:\n",
    "        druha_osa = 1\n",
    "        druha_osa_y = {\n",
    "            \"title\": {\"text\": osay2},\n",
    "            \"opposite\": True,\n",
    "            \"max\": naproti[0].max(),\n",
    "            \"min\": 0,\n",
    "        }\n",
    "        nastaveni[\"yAxis\"].append(druha_osa_y)\n",
    "\n",
    "    my_chart = Chart(container=target, options=nastaveni)\n",
    "\n",
    "    procenta = [p.name for p in procenta]\n",
    "    naproti = [n.name for n in naproti]\n",
    "    skryte = [s.name for s in skryte]\n",
    "\n",
    "    def vykresleni(serie, typ):\n",
    "\n",
    "        pocitadlo_barev = 0\n",
    "        \n",
    "        for s in serie:\n",
    "            popisek = s.name\n",
    "            print(popisek)\n",
    "\n",
    "            if s.name in skryte:\n",
    "                viditelnost = False\n",
    "            else:\n",
    "                viditelnost = True\n",
    "\n",
    "            ktera_osa = 0\n",
    "            if s.name in naproti:\n",
    "                ktera_osa = druha_osa\n",
    "\n",
    "            if s.name in procenta:\n",
    "                s = [round(x * 100, zaokrouhleni) for x in s.fillna(0).to_list()]\n",
    "                my_chart.add_series(\n",
    "                    typ(\n",
    "                        data=s,\n",
    "                        visible=viditelnost,\n",
    "                        name=popisek,\n",
    "                        y_axis=druha_osa,\n",
    "                        tooltip={\"valueSuffix\": \" %\"},\n",
    "                    )\n",
    "                )\n",
    "\n",
    "            if any(s.name == x.name for x in rozpeti):\n",
    "                line_id = f\"line_{popisek}\"\n",
    "                print(line_id)\n",
    "                my_chart.add_series(\n",
    "                    LineSeries(\n",
    "                        data=s.apply(lambda x: x[2]).fillna(0).to_list(),\n",
    "                        visible=viditelnost,\n",
    "                        id = line_id,\n",
    "                        name=popisek,\n",
    "                        y_axis=ktera_osa,\n",
    "                        color = barvy[pocitadlo_barev],\n",
    "                        tooltip={\"valuePrefix\": \"střední: \",\"valueSuffix\": \" Kč\"},\n",
    "                    ))\n",
    "                my_chart.add_series(\n",
    "                    AreaRangeSeries(\n",
    "                        data=s.apply(lambda x: x[0:2]).fillna(0).to_list(),\n",
    "                        visible=viditelnost,\n",
    "                        type='arearange',\n",
    "                        name=popisek,\n",
    "                        linkedto = line_id,\n",
    "                        y_axis=ktera_osa,\n",
    "                        color = barvy[pocitadlo_barev],\n",
    "                        fillOpacity=0.3,\n",
    "                        tooltip={\"valueSuffix\": \" Kč\"},\n",
    "                    lineWidth=0,\n",
    "                    marker={\"enabled\": False}\n",
    "                    ))\n",
    "                pocitadlo_barev += 1\n",
    "\n",
    "   #         else:\n",
    "                #my_chart.add_series(\n",
    "                    #typ(\n",
    "                     #   data=s.fillna(0).to_list(),\n",
    "                    #    visible=viditelnost,\n",
    "                   #     name=popisek,\n",
    "                  #      y_axis=ktera_osa,\n",
    "                 #   )\n",
    "                #)\n",
    "\n",
    "    if len(sloupcovy) > 0:\n",
    "        vykresleni(sloupcovy, ColumnSeries)\n",
    "    if len(carovy) > 0:\n",
    "        vykresleni(carovy, LineSeries)\n",
    "    if len(vodorovny) > 0:\n",
    "        vykresleni(vodorovny, BarSeries)\n",
    "    if len(rozpeti) > 0:\n",
    "        vykresleni(rozpeti, AreaRangeSeries)\n",
    "\n",
    "    my_chart.options.colors = barvy\n",
    "\n",
    "    if naopak:\n",
    "        my_chart.options.legend = Legend(reversed=True)\n",
    "\n",
    "    my_chart.options.title = Title(text=titulek, align=\"left\", margin=30)\n",
    "\n",
    "    if len(podtitulek) > 0:\n",
    "        my_chart.options.subtitle = Subtitle(text=podtitulek, align=\"left\")\n",
    "\n",
    "    my_chart.options.credits = Credits(text=kredity[0], enabled=True, href=kredity[1])\n",
    "\n",
    "    as_js_literal = my_chart.to_js_literal()\n",
    "\n",
    "    if skrytnuly == True:\n",
    "        as_js_literal = as_js_literal.replace(\"y: 0.0\", \"y: null\")\n",
    "\n",
    " #   if len(rozpeti) > 0:\n",
    "    \n",
    "  #      as_js_literal = as_js_literal.replace(\n",
    "   #         '\"type\":\"arearange\"', \n",
    "    #        '\"type\":\"arearange\", \"linkedTo\": \"previous\", \"fillOpacity\": 0.3, \"lineWidth\": 0, \"marker\": {\"enabled\": false}'\n",
    "     #   )\n",
    "\n",
    "    \n",
    "#    if len(rozpeti) > 0:\n",
    "#        as_js_literal = as_js_literal.replace(\"type: 'arearange'\", \"type: 'arearange',linkedTo: 'previous',fillOpacity: 0.3,lineWidth: 0,marker: {enabled: false}\")\n",
    "\n",
    "    as_js_literal = as_js_literal.splitlines()\n",
    "\n",
    "    as_js_literal2 = []\n",
    "    \n",
    "    for line in as_js_literal:\n",
    "        if 'id: ' in line:\n",
    "            ajdy = line.split(\"'\")[1]\n",
    "            print(ajdy)\n",
    "        if \"type: 'arearange'\" in line:\n",
    "            line = line.replace(\" type: 'arearange'\",f\" type: 'arearange',linkedTo: '{ajdy}', fillOpacity: 0.3, lineWidth: 0\")\n",
    "        as_js_literal2.append(line)\n",
    "    as_js_literal2 = \"\\n\".join(as_js_literal2)\n",
    "    \n",
    "    code = f\"<html><head><title>{titulek}</title></head><body>{pred}{as_js_literal2}</script></body></html>\"\n",
    "\n",
    "    \n",
    "    if not os.path.exists(\"grafy\"):\n",
    "        os.mkdir(\"grafy\")\n",
    "\n",
    "    with open(os.path.join(\"grafy\", target + \".html\"), \"w+\") as f:\n",
    "        f.write(code)\n",
    "\n",
    "    with open(os.path.join(\"grafy\", target + \".txt\"), \"w+\") as f:\n",
    "        f.write(f\"{pred}{as_js_literal2}</script>\")\n",
    "\n",
    "        print(\"Graf uložen.\")"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 34,
   "id": "6cac8198-ad17-4420-9410-401eb9fd3126",
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "ČD Praha &#8596; Varšava\n",
      "line_ČD Praha &#8596; Varšava\n",
      "ČD Praha &#8596; Berlín\n",
      "line_ČD Praha &#8596; Berlín\n",
      "ČD Praha &#8596; Bratislava\n",
      "line_ČD Praha &#8596; Bratislava\n",
      "RJ Praha &#8596; Bratislava\n",
      "line_RJ Praha &#8596; Bratislava\n",
      "ČD Praha &#8596; Vídeň\n",
      "line_ČD Praha &#8596; Vídeň\n",
      "RJ Praha &#8596; Vídeň\n",
      "line_RJ Praha &#8596; Vídeň\n",
      "LE Pardubice &#8596; Košice\n",
      "line_LE Pardubice &#8596; Košice\n",
      "RJ Praha &#8596; Čop\n",
      "line_RJ Praha &#8596; Čop\n",
      "ČD Praha &#8596; Budapešť\n",
      "line_ČD Praha &#8596; Budapešť\n",
      "RJ Praha &#8596; Budapešť\n",
      "line_RJ Praha &#8596; Budapešť\n",
      "ČD Ústí nad Labem &#8596; Drážďany\n",
      "line_ČD Ústí nad Labem &#8596; Drážďany\n",
      "LE Praha &#8596; Krakov\n",
      "line_LE Praha &#8596; Krakov\n",
      "line_ČD Praha &#8596; Varšava\n",
      "line_ČD Praha &#8596; Berlín\n",
      "line_ČD Praha &#8596; Bratislava\n",
      "line_RJ Praha &#8596; Bratislava\n",
      "line_ČD Praha &#8596; Vídeň\n",
      "line_RJ Praha &#8596; Vídeň\n",
      "line_LE Pardubice &#8596; Košice\n",
      "line_RJ Praha &#8596; Čop\n",
      "line_ČD Praha &#8596; Budapešť\n",
      "line_RJ Praha &#8596; Budapešť\n",
      "line_ČD Ústí nad Labem &#8596; Drážďany\n",
      "line_LE Praha &#8596; Krakov\n",
      "Graf uložen.\n"
     ]
    }
   ],
   "source": [
    "irozhlas_graf(rozpeti=ukazat, skryte=ukazat[1:], barvy=farby, target='mezistatni_jizdne', titulek='Mezistátní jízdné podle předstihu nákupu', osaymin=0, podtitulek='Nejnižší, střední a nejvyšší cena na přímých spojích. Od 14. dne dál se do dat částečně promítají prosincové změny jízdních řádů a zdražení některých spojů.', kredity=['Zdroj dat: e-shopy dopravců od 7. 11. do 1. 12. 2024. Vizualizace: iROZHLAS.cz','https://www.irozhlas.cz/zpravy-tag/datova-zurnalistika'])"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3 (ipykernel)",
   "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.12.3"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 5
}