1_simple_flag.ipynb

{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### 1. Вводные функции\n",
    "\n",
    "Существует много библиотек, нацеленных на обработку фида GTFS и его анализа, однако считаю, что достаточно имплементировать несколько функций, занимающихся \"вытягиванием\" данных из классов GTFS в формате табличной структуры Dataframe.\n",
    "\n",
    "Сперва необходимо импортировать все таблицы в dataframe:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import pandas as pd\n",
    "folder='k26_static'\n",
    "\n",
    "agency=pd.read_csv(folder+r'\\agency.txt', sep=',')\n",
    "calendar=pd.read_csv(folder+r'\\calendar.txt', sep=',')\n",
    "calendar_dates=pd.read_csv(folder+r'\\calendar_dates.txt', sep=',')\n",
    "routes=pd.read_csv(folder+r'\\routes.txt', sep=',')\n",
    "trips=pd.read_csv(folder+r'\\trips.txt', sep=',')\n",
    "stops=pd.read_csv(folder+r'\\stops.txt', sep=',')\n",
    "stop_times=pd.read_csv(folder+r'\\stop_times.txt', sep=',')\n",
    "shapes=pd.read_csv(folder+r'\\shapes.txt', sep=',')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Вывести информацию по одной из остановок по идентификатору можно следующим образом:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "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>stop_id</th>\n",
       "      <th>stop_code</th>\n",
       "      <th>stop_name</th>\n",
       "      <th>stop_desc</th>\n",
       "      <th>stop_lat</th>\n",
       "      <th>stop_lon</th>\n",
       "      <th>zone_id</th>\n",
       "      <th>stop_url</th>\n",
       "      <th>location_type</th>\n",
       "      <th>parent_station</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>14</th>\n",
       "      <td>21</td>\n",
       "      <td>NaN</td>\n",
       "      <td>Магазин \"Аквариум\"</td>\n",
       "      <td>NaN</td>\n",
       "      <td>56.256405</td>\n",
       "      <td>93.533688</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "    stop_id  stop_code           stop_name  stop_desc   stop_lat   stop_lon  \\\n",
       "14       21        NaN  Магазин \"Аквариум\"        NaN  56.256405  93.533688   \n",
       "\n",
       "    zone_id  stop_url  location_type  parent_station  \n",
       "14      NaN       NaN            NaN             NaN  "
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "stop_id=21\n",
    "stop=stops[stops.stop_id==stop_id]\n",
    "stop"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Чтобы вывести информацию по всем прибытиям на остановку в фиде, можно обратиться к классу stop_times.\n",
    "Вместе с каждым прибытием на остановку можно сразу выводить информацию о поездке каждого из прибытий с помощью связи таблиц по идентификатору trip_id:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "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>trip_id</th>\n",
       "      <th>arrival_time</th>\n",
       "      <th>departure_time</th>\n",
       "      <th>stop_id</th>\n",
       "      <th>stop_sequence</th>\n",
       "      <th>stop_headsign</th>\n",
       "      <th>pickup_type</th>\n",
       "      <th>drop_off_type</th>\n",
       "      <th>shape_dist_traveled</th>\n",
       "      <th>route_id</th>\n",
       "      <th>service_id</th>\n",
       "      <th>trip_headsign</th>\n",
       "      <th>direction_id</th>\n",
       "      <th>block_id</th>\n",
       "      <th>shape_id</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>bus_2_dir0_05_50_winter_weekday</td>\n",
       "      <td>05:53:00</td>\n",
       "      <td>05:53:00</td>\n",
       "      <td>21</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1</th>\n",
       "      <td>bus_2_dir0_06_20_winter_weekday</td>\n",
       "      <td>06:23:00</td>\n",
       "      <td>06:23:00</td>\n",
       "      <td>21</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>2</th>\n",
       "      <td>bus_2_dir0_07_05_winter_weekday</td>\n",
       "      <td>07:08:00</td>\n",
       "      <td>07:08:00</td>\n",
       "      <td>21</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>bus_2_dir0_07_26_winter_weekday</td>\n",
       "      <td>07:29:00</td>\n",
       "      <td>07:29:00</td>\n",
       "      <td>21</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>bus_2_dir0_08_00_winter_weekday</td>\n",
       "      <td>08:03:00</td>\n",
       "      <td>08:03:00</td>\n",
       "      <td>21</td>\n",
       "      <td>4</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</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",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>446</th>\n",
       "      <td>bus_6_dir1_18_38_0</td>\n",
       "      <td>18:44:00</td>\n",
       "      <td>18:44:00</td>\n",
       "      <td>21</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>447</th>\n",
       "      <td>bus_6_dir1_18_58_0</td>\n",
       "      <td>19:04:00</td>\n",
       "      <td>19:04:00</td>\n",
       "      <td>21</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>448</th>\n",
       "      <td>bus_6_dir1_19_38_0</td>\n",
       "      <td>19:44:00</td>\n",
       "      <td>19:44:00</td>\n",
       "      <td>21</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>449</th>\n",
       "      <td>bus_6_dir1_20_18_0</td>\n",
       "      <td>20:24:00</td>\n",
       "      <td>20:24:00</td>\n",
       "      <td>21</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>450</th>\n",
       "      <td>bus_6_dir1_21_08_0</td>\n",
       "      <td>21:14:00</td>\n",
       "      <td>21:14:00</td>\n",
       "      <td>21</td>\n",
       "      <td>5</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>NaN</td>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>451 rows × 15 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "                             trip_id arrival_time departure_time  stop_id  \\\n",
       "0    bus_2_dir0_05_50_winter_weekday     05:53:00       05:53:00       21   \n",
       "1    bus_2_dir0_06_20_winter_weekday     06:23:00       06:23:00       21   \n",
       "2    bus_2_dir0_07_05_winter_weekday     07:08:00       07:08:00       21   \n",
       "3    bus_2_dir0_07_26_winter_weekday     07:29:00       07:29:00       21   \n",
       "4    bus_2_dir0_08_00_winter_weekday     08:03:00       08:03:00       21   \n",
       "..                               ...          ...            ...      ...   \n",
       "446               bus_6_dir1_18_38_0     18:44:00       18:44:00       21   \n",
       "447               bus_6_dir1_18_58_0     19:04:00       19:04:00       21   \n",
       "448               bus_6_dir1_19_38_0     19:44:00       19:44:00       21   \n",
       "449               bus_6_dir1_20_18_0     20:24:00       20:24:00       21   \n",
       "450               bus_6_dir1_21_08_0     21:14:00       21:14:00       21   \n",
       "\n",
       "     stop_sequence  stop_headsign  pickup_type  drop_off_type  \\\n",
       "0                4            NaN          NaN            NaN   \n",
       "1                4            NaN          NaN            NaN   \n",
       "2                4            NaN          NaN            NaN   \n",
       "3                4            NaN          NaN            NaN   \n",
       "4                4            NaN          NaN            NaN   \n",
       "..             ...            ...          ...            ...   \n",
       "446              5            NaN          NaN            NaN   \n",
       "447              5            NaN          NaN            NaN   \n",
       "448              5            NaN          NaN            NaN   \n",
       "449              5            NaN          NaN            NaN   \n",
       "450              5            NaN          NaN            NaN   \n",
       "\n",
       "     shape_dist_traveled route_id      service_id trip_headsign  direction_id  \\\n",
       "0                    NaN    bus_2  winter_weekday         КПП-1             1   \n",
       "1                    NaN    bus_2  winter_weekday         КПП-1             1   \n",
       "2                    NaN    bus_2  winter_weekday         КПП-1             1   \n",
       "3                    NaN    bus_2  winter_weekday         КПП-1             1   \n",
       "4                    NaN    bus_2  winter_weekday         КПП-1             1   \n",
       "..                   ...      ...             ...           ...           ...   \n",
       "446                  NaN    bus_6               0          ПАТП             0   \n",
       "447                  NaN    bus_6               0          ПАТП             0   \n",
       "448                  NaN    bus_6               0          ПАТП             0   \n",
       "449                  NaN    bus_6               0          ПАТП             0   \n",
       "450                  NaN    bus_6               0          ПАТП             0   \n",
       "\n",
       "       block_id          shape_id  \n",
       "0    bus_2_veh0  bus_2_dir0_shape  \n",
       "1    bus_2_veh0  bus_2_dir0_shape  \n",
       "2    bus_2_veh0  bus_2_dir0_shape  \n",
       "3    bus_2_veh0  bus_2_dir0_shape  \n",
       "4    bus_2_veh0  bus_2_dir0_shape  \n",
       "..          ...               ...  \n",
       "446  bus_6_veh1  bus_6_dir1_shape  \n",
       "447  bus_6_veh1  bus_6_dir1_shape  \n",
       "448  bus_6_veh1  bus_6_dir1_shape  \n",
       "449  bus_6_veh1  bus_6_dir1_shape  \n",
       "450  bus_6_veh1  bus_6_dir1_shape  \n",
       "\n",
       "[451 rows x 15 columns]"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "st_selection=stop_times[stop_times.stop_id==stop_id]\n",
    "st_selection_merge=st_selection.merge(trips, on='trip_id', how='left')\n",
    "# пока что на данном этапе нет вывода информации по конкретному дню недели или делению на будний или выходной день\n",
    "st_selection_merge"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Только поездки: "
   ]
  },
  {
   "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>route_id</th>\n",
       "      <th>service_id</th>\n",
       "      <th>trip_id</th>\n",
       "      <th>trip_headsign</th>\n",
       "      <th>direction_id</th>\n",
       "      <th>block_id</th>\n",
       "      <th>shape_id</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>250</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>bus_2_dir0_05_50_winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>251</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>bus_2_dir0_06_20_winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>252</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>bus_2_dir0_07_05_winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>253</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>bus_2_dir0_07_26_winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>254</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>winter_weekday</td>\n",
       "      <td>bus_2_dir0_08_00_winter_weekday</td>\n",
       "      <td>КПП-1</td>\n",
       "      <td>1</td>\n",
       "      <td>bus_2_veh0</td>\n",
       "      <td>bus_2_dir0_shape</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",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1511</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_dir1_18_38_0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1512</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_dir1_18_58_0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1513</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_dir1_19_38_0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1514</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_dir1_20_18_0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>1515</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_dir1_21_08_0</td>\n",
       "      <td>ПАТП</td>\n",
       "      <td>0</td>\n",
       "      <td>bus_6_veh1</td>\n",
       "      <td>bus_6_dir1_shape</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "<p>451 rows × 7 columns</p>\n",
       "</div>"
      ],
      "text/plain": [
       "     route_id      service_id                          trip_id trip_headsign  \\\n",
       "250     bus_2  winter_weekday  bus_2_dir0_05_50_winter_weekday         КПП-1   \n",
       "251     bus_2  winter_weekday  bus_2_dir0_06_20_winter_weekday         КПП-1   \n",
       "252     bus_2  winter_weekday  bus_2_dir0_07_05_winter_weekday         КПП-1   \n",
       "253     bus_2  winter_weekday  bus_2_dir0_07_26_winter_weekday         КПП-1   \n",
       "254     bus_2  winter_weekday  bus_2_dir0_08_00_winter_weekday         КПП-1   \n",
       "...       ...             ...                              ...           ...   \n",
       "1511    bus_6               0               bus_6_dir1_18_38_0          ПАТП   \n",
       "1512    bus_6               0               bus_6_dir1_18_58_0          ПАТП   \n",
       "1513    bus_6               0               bus_6_dir1_19_38_0          ПАТП   \n",
       "1514    bus_6               0               bus_6_dir1_20_18_0          ПАТП   \n",
       "1515    bus_6               0               bus_6_dir1_21_08_0          ПАТП   \n",
       "\n",
       "      direction_id    block_id          shape_id  \n",
       "250              1  bus_2_veh0  bus_2_dir0_shape  \n",
       "251              1  bus_2_veh0  bus_2_dir0_shape  \n",
       "252              1  bus_2_veh0  bus_2_dir0_shape  \n",
       "253              1  bus_2_veh0  bus_2_dir0_shape  \n",
       "254              1  bus_2_veh0  bus_2_dir0_shape  \n",
       "...            ...         ...               ...  \n",
       "1511             0  bus_6_veh1  bus_6_dir1_shape  \n",
       "1512             0  bus_6_veh1  bus_6_dir1_shape  \n",
       "1513             0  bus_6_veh1  bus_6_dir1_shape  \n",
       "1514             0  bus_6_veh1  bus_6_dir1_shape  \n",
       "1515             0  bus_6_veh1  bus_6_dir1_shape  \n",
       "\n",
       "[451 rows x 7 columns]"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "trip_ids=list(set(st_selection.trip_id))\n",
    "t_selection=trips[trips.trip_id.isin(trip_ids)]\n",
    "t_selection"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Какие маршруты проходят через остановку?"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 5,
   "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>route_id</th>\n",
       "      <th>agency_id</th>\n",
       "      <th>route_short_name</th>\n",
       "      <th>route_long_name</th>\n",
       "      <th>route_desc</th>\n",
       "      <th>route_type</th>\n",
       "      <th>route_url</th>\n",
       "      <th>route_color</th>\n",
       "      <th>route_text_color</th>\n",
       "    </tr>\n",
       "  </thead>\n",
       "  <tbody>\n",
       "    <tr>\n",
       "      <th>0</th>\n",
       "      <td>bus_6</td>\n",
       "      <td>0</td>\n",
       "      <td>6</td>\n",
       "      <td>ПАТП - Лукаши</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>#8dc73f</td>\n",
       "      <td>#ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>3</th>\n",
       "      <td>bus_2</td>\n",
       "      <td>0</td>\n",
       "      <td>2</td>\n",
       "      <td>Ленина 75 - КПП-1</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>#6ecff6</td>\n",
       "      <td>#ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>4</th>\n",
       "      <td>bus_9</td>\n",
       "      <td>0</td>\n",
       "      <td>9</td>\n",
       "      <td>Склады - Маяк</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>#ffcb04</td>\n",
       "      <td>#ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>6</th>\n",
       "      <td>bus_10</td>\n",
       "      <td>0</td>\n",
       "      <td>10</td>\n",
       "      <td>Ленина 75 - Мира</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>#a14c0c</td>\n",
       "      <td>#ffffff</td>\n",
       "    </tr>\n",
       "    <tr>\n",
       "      <th>7</th>\n",
       "      <td>bus_44</td>\n",
       "      <td>0</td>\n",
       "      <td>44</td>\n",
       "      <td>Ленина 75 - Церковь</td>\n",
       "      <td>NaN</td>\n",
       "      <td>3</td>\n",
       "      <td>NaN</td>\n",
       "      <td>#f47920</td>\n",
       "      <td>#ffffff</td>\n",
       "    </tr>\n",
       "  </tbody>\n",
       "</table>\n",
       "</div>"
      ],
      "text/plain": [
       "  route_id  agency_id  route_short_name      route_long_name  route_desc  \\\n",
       "0    bus_6          0                 6        ПАТП - Лукаши         NaN   \n",
       "3    bus_2          0                 2    Ленина 75 - КПП-1         NaN   \n",
       "4    bus_9          0                 9        Склады - Маяк         NaN   \n",
       "6   bus_10          0                10     Ленина 75 - Мира         NaN   \n",
       "7   bus_44          0                44  Ленина 75 - Церковь         NaN   \n",
       "\n",
       "   route_type  route_url route_color route_text_color  \n",
       "0           3        NaN     #8dc73f          #ffffff  \n",
       "3           3        NaN     #6ecff6          #ffffff  \n",
       "4           3        NaN     #ffcb04          #ffffff  \n",
       "6           3        NaN     #a14c0c          #ffffff  \n",
       "7           3        NaN     #f47920          #ffffff  "
      ]
     },
     "execution_count": 5,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "route_ids=list(set(t_selection.route_id))\n",
    "r_selection=routes[routes.route_id.isin(route_ids)]\n",
    "r_selection"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Теперь задача сложнее: нам нужно узнать первый и последний рейсы каждого из маршрутов, проходящих через остановку. На данном этапе - без учета по конкретным дням недели или делению на будний или выходной день"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 6,
   "metadata": {},
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Маршрут 6 до ПАТП: 06:59:00-21:14:00\n",
      "Маршрут 2 до КПП-1: 05:53:00-21:53:00\n",
      "Маршрут 9 до Маяк: 06:09:00-21:09:00\n",
      "Маршрут 10 до Мира: 05:48:00-22:33:00\n",
      "Маршрут 44 до Церковь: 06:18:00-22:14:00\n"
     ]
    }
   ],
   "source": [
    "route_ids=list(r_selection.route_id)\n",
    "for route_id in route_ids:\n",
    "    route=r_selection[r_selection.route_id==route_id]\n",
    "    route_arrives=st_selection_merge[st_selection_merge.route_id==route_id]\n",
    "    \n",
    "    # Без учета календаря\n",
    "    max=route_arrives.arrival_time.max()\n",
    "    min=route_arrives.arrival_time.min()\n",
    "    # Без учета возможных укоротов/вариантов\n",
    "    main_headsign=route_arrives.trip_headsign.mode().item()\n",
    "    print(f'Маршрут {route['route_short_name'].item()} до {main_headsign}: {min}-{max}')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Мы получили всё для того, чтобы получить данные для визуализации флага остановки. Обернем все в словарь и дополнительно укажем цвет маршрута и цвет текста маршрута:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 7,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "{'bus_6': {'start': '06:59:00',\n",
       "  'end': '21:14:00',\n",
       "  'headsign': 'ПАТП',\n",
       "  'route_color': '#8dc73f',\n",
       "  'route_text_color': '#ffffff',\n",
       "  'route_short_name': 6},\n",
       " 'bus_2': {'start': '05:53:00',\n",
       "  'end': '21:53:00',\n",
       "  'headsign': 'КПП-1',\n",
       "  'route_color': '#6ecff6',\n",
       "  'route_text_color': '#ffffff',\n",
       "  'route_short_name': 2},\n",
       " 'bus_9': {'start': '06:09:00',\n",
       "  'end': '21:09:00',\n",
       "  'headsign': 'Маяк',\n",
       "  'route_color': '#ffcb04',\n",
       "  'route_text_color': '#ffffff',\n",
       "  'route_short_name': 9},\n",
       " 'bus_10': {'start': '05:48:00',\n",
       "  'end': '22:33:00',\n",
       "  'headsign': 'Мира',\n",
       "  'route_color': '#a14c0c',\n",
       "  'route_text_color': '#ffffff',\n",
       "  'route_short_name': 10},\n",
       " 'bus_44': {'start': '06:18:00',\n",
       "  'end': '22:14:00',\n",
       "  'headsign': 'Церковь',\n",
       "  'route_color': '#f47920',\n",
       "  'route_text_color': '#ffffff',\n",
       "  'route_short_name': 44}}"
      ]
     },
     "execution_count": 7,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "route_ids=list(r_selection.route_id)\n",
    "route_info=dict()\n",
    "for route_id in route_ids:\n",
    "    route=r_selection[r_selection.route_id==route_id]\n",
    "    route_arrives=st_selection_merge[st_selection_merge.route_id==route_id]\n",
    "    \n",
    "    # Без учета календаря\n",
    "    max=route_arrives.arrival_time.max()\n",
    "    min=route_arrives.arrival_time.min()\n",
    "    # Без учета возможных укоротов/вариантов\n",
    "    main_headsign=route_arrives.trip_headsign.mode().item()\n",
    "    route_info[route_id]={'start': min, 'end': max, 'headsign': main_headsign,\n",
    "                                  'route_color': route['route_color'].item(), \n",
    "                                  'route_text_color': route['route_text_color'].item(), 'route_short_name': route['route_short_name'].item()}\n",
    "route_info"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Вся последовательность действий обернута в класс Feed в скрипте feeds.py. \n",
    "\n",
    "Теперь можно приступить непосредственно к визуализации флага:"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 8,
   "metadata": {},
   "outputs": [
    {
     "name": "stderr",
     "output_type": "stream",
     "text": [
      "c:\\Users\\gamma\\AppData\\Local\\Programs\\Python\\Python312\\Lib\\site-packages\\fpdf\\__init__.py:39: UserWarning: You have both PyFPDF & fpdf2 installed. Both packages cannot be installed at the same time as they share the same module namespace. To only keep fpdf2, run: pip uninstall --yes pypdf && pip install --upgrade fpdf2\n",
      "  warnings.warn(\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:20: DeprecationWarning: \"uni\" parameter is deprecated since v2.5.1, unused and will soon be removed\n",
      "  pdf.add_font('MoscowSans', '', r\"D:\\fonts\\Moscow Sans\\MoscowSans-Regular.otf\", uni=True)\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:36: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=1 use new_x=XPos.LMARGIN, new_y=YPos.NEXT.\n",
      "  pdf.cell(190, 20, 'Маршруты', 0, 1, 'L')\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:44: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=0 use new_x=XPos.RIGHT, new_y=YPos.TOP.\n",
      "  pdf.cell(30, 20, str(stop[route]['route_short_name']), 0, 0, 'C', True)\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:46: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=0 use new_x=XPos.RIGHT, new_y=YPos.TOP.\n",
      "  pdf.cell(5, 10, '', 0, 0, 'L', False)\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:50: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=0 use new_x=XPos.RIGHT, new_y=YPos.TOP.\n",
      "  pdf.cell(160, 15, str(stop[route]['headsign']), 0, 0, 'L', False)\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:53: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=0 use new_x=XPos.RIGHT, new_y=YPos.TOP.\n",
      "  pdf.cell(35, 20, '', 0, 0, 'L', False)\n",
      "C:\\Users\\gamma\\AppData\\Local\\Temp\\ipykernel_2780\\3951510542.py:56: DeprecationWarning: The parameter \"ln\" is deprecated since v2.5.2. Instead of ln=0 use new_x=XPos.RIGHT, new_y=YPos.TOP.\n",
      "  pdf.cell(160, 5, ServiceTime(stop[route]['start'],stop[route]['end']), 0, 0, 'L', False)\n"
     ]
    },
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "Stop name Магазин \"Аквариум\"\n",
      "Serves 5 routes: [6, 2, 9, 10, 44]\n",
      "{'bus_2': {'start': '05:53:00', 'end': '21:53:00', 'headsign': 'КПП-1', 'route_color': '#6ecff6', 'route_text_color': '#ffffff', 'route_short_name': 2}, 'bus_6': {'start': '06:59:00', 'end': '21:14:00', 'headsign': 'ПАТП', 'route_color': '#8dc73f', 'route_text_color': '#ffffff', 'route_short_name': 6}, 'bus_9': {'start': '06:09:00', 'end': '21:09:00', 'headsign': 'Маяк', 'route_color': '#ffcb04', 'route_text_color': '#ffffff', 'route_short_name': 9}, 'bus_10': {'start': '05:48:00', 'end': '22:33:00', 'headsign': 'Мира', 'route_color': '#a14c0c', 'route_text_color': '#ffffff', 'route_short_name': 10}, 'bus_44': {'start': '06:18:00', 'end': '22:14:00', 'headsign': 'Церковь', 'route_color': '#f47920', 'route_text_color': '#ffffff', 'route_short_name': 44}}\n"
     ]
    }
   ],
   "source": [
    "from fpdf import FPDF\n",
    "\n",
    "from feed import Feed\n",
    "\n",
    "def HexToRGB(h):\n",
    "    # Перевод HTML цвета в RGB\n",
    "    h = h.lstrip('#')\n",
    "    return tuple(int(h[i:i+2], 16) for i in (0, 2, 4))\n",
    "def ServiceTime(start, end):\n",
    "    # Исключение секунд в времени работы маршрута\n",
    "    return f\"{start.split(':')[0]}:{start.split(':')[1]}...{end.split(':')[0]}:{end.split(':')[1]}\"\n",
    "\n",
    "stop_id=21\n",
    "feed=Feed('k26_static')\n",
    "stop=feed.DescribeForFlag(stop_id)\n",
    "stop_name=feed.StopInfo(stop_id).stop_name.item()\n",
    "print(stop)\n",
    "# Объявление класса PDF-файла\n",
    "pdf = FPDF(format='A4')\n",
    "pdf.add_font('MoscowSans', '', r\"D:\\fonts\\Moscow Sans\\MoscowSans-Regular.otf\", uni=True)\n",
    "pdf.alias_nb_pages()\n",
    "# Добавляем страницу\n",
    "pdf.add_page()\n",
    "# Перед тем, как добавлять текстовый блок, необходимо задать параметры: шрифт, вес, кегль\n",
    "pdf.set_font('MoscowSans', '', 40)\n",
    "# Добавляем пиктограмму: координаты точки крепления x,y и размер пиктограммы w,h \n",
    "pdf.image(\"stop-01.png\", x=10, y=10, w = 50, h = 50)\n",
    "# Отступ от картинки для нового текстового блока\n",
    "pdf.cell(60)\n",
    "# Текстовый блок с автопереносом: размер блока, текст, обводка, центрирование\n",
    "pdf.multi_cell(120, 30, stop_name, 0, 'L')\n",
    "# Меняем размер шрифта\n",
    "pdf.set_font('MoscowSans', '', 32)\n",
    "# Переход на новую строку\n",
    "pdf.ln(h=20)\n",
    "pdf.cell(190, 20, 'Маршруты', 0, 1, 'L')\n",
    "pdf.ln(h=5)\n",
    "# Для каждого маршрута, проходящего через остановку, визуализируем информацию\n",
    "for route in list(stop.keys()):\n",
    "    pdf.set_font('MoscowSans', '', 40)\n",
    "    # Объявляем цвет фона текстового блока и цвет текста в нем\n",
    "    pdf.set_fill_color(*HexToRGB(stop[route]['route_color']))\n",
    "    pdf.set_text_color(*HexToRGB(stop[route]['route_text_color']))\n",
    "    pdf.cell(30, 20, str(stop[route]['route_short_name']), 0, 0, 'C', True)\n",
    "    # Отступы\n",
    "    pdf.cell(5, 10, '', 0, 0, 'L', False)\n",
    "    # Черный цвет текста\n",
    "    pdf.set_text_color(r=0)\n",
    "    # Направление маршрута\n",
    "    pdf.cell(160, 15, str(stop[route]['headsign']), 0, 0, 'L', False)\n",
    "    pdf.ln(h=15)\n",
    "    # Отступы\n",
    "    pdf.cell(35, 20, '', 0, 0, 'L', False)\n",
    "    pdf.set_font('MoscowSans', '', 24)\n",
    "    # Время первого и последнего рейса\n",
    "    pdf.cell(160, 5, ServiceTime(stop[route]['start'],stop[route]['end']), 0, 0, 'L', False)\n",
    "    pdf.ln(h=15)\n",
    "\n",
    "# Экспортируем результат\n",
    "pdf.output(f'1_Extract.pdf')"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "Скрипт можно дополнять бесконечно, делая его приятнее человеческому глазу. Но главное - не нужно в векторном редакторе своими руками всё настраивать!"
   ]
  }
 ],
 "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.12.1"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}