GTFS Loader Example¶

In [1]:

from pathlib import Path
from srai.loaders import GTFSLoader, download_file
import gtfs_kit as gk
import geopandas as gpd
from shapely.geometry import Point
from srai.constants import WGS84_CRS

from pathlib import Path from srai.loaders import GTFSLoader, download_file import gtfs_kit as gk import geopandas as gpd from shapely.geometry import Point from srai.constants import WGS84_CRS

Download an example GTFS feed from Wroclaw, Poland¶

In this notebook we use the GTFS feed for Wroclaw, Poland as an example, which is available in Wroclaw's open data repository[1]. This download uses transitfeeds.com[2] to download the feed, but you can also download the feed directly from the Wroclaw open data repository.

1. https://www.wroclaw.pl/open-data/dataset/rozkladjazdytransportupublicznegoplik_data
2. https://transitfeeds.com/p/mpk-wroc-aw/663

In [2]:

wroclaw_gtfs = Path().resolve() / "files" / "example.zip"
gtfs_url = "https://transitfeeds.com/p/mpk-wroc-aw/663/20221221/download"

download_file(gtfs_url, wroclaw_gtfs.as_posix())

wroclaw_gtfs = Path().resolve() / "files" / "example.zip" gtfs_url = "https://transitfeeds.com/p/mpk-wroc-aw/663/20221221/download" download_file(gtfs_url, wroclaw_gtfs.as_posix())

example.zip: 100%|██████████| 13.7M/13.7M [00:00<00:00, 19.2MiB/s]

Peek at the feed using gtfs_kit directly¶

In [3]:

feed = gk.read_feed(wroclaw_gtfs, dist_units="km")

stops_df = feed.stops[["stop_id", "stop_lat", "stop_lon"]].set_index("stop_id")
stops_df["geometry"] = stops_df.apply(lambda row: Point(row["stop_lon"], row["stop_lat"]), axis=1)

stops_gdf = gpd.GeoDataFrame(
    stops_df,
    geometry="geometry",
    crs=WGS84_CRS,
)

stops_gdf.explore(tiles="CartoDB positron")

feed = gk.read_feed(wroclaw_gtfs, dist_units="km") stops_df = feed.stops[["stop_id", "stop_lat", "stop_lon"]].set_index("stop_id") stops_df["geometry"] = stops_df.apply(lambda row: Point(row["stop_lon"], row["stop_lat"]), axis=1) stops_gdf = gpd.GeoDataFrame( stops_df, geometry="geometry", crs=WGS84_CRS, ) stops_gdf.explore(tiles="CartoDB positron")

Out[3]:

Make this Notebook Trusted to load map: File -> Trust Notebook

Use GTFSLoader to load stops statistics from the feed¶

In [4]:

gtfs_loader = GTFSLoader()
trips_gdf = gtfs_loader.load(wroclaw_gtfs)

print(trips_gdf.columns)

gtfs_loader = GTFSLoader() trips_gdf = gtfs_loader.load(wroclaw_gtfs) print(trips_gdf.columns)

Index(['trips_at_0', 'trips_at_1', 'trips_at_2', 'trips_at_3', 'trips_at_4',
       'trips_at_5', 'trips_at_6', 'trips_at_7', 'trips_at_8', 'trips_at_9',
       'trips_at_10', 'trips_at_11', 'trips_at_12', 'trips_at_13',
       'trips_at_14', 'trips_at_15', 'trips_at_16', 'trips_at_17',
       'trips_at_18', 'trips_at_19', 'trips_at_20', 'trips_at_21',
       'trips_at_22', 'trips_at_23', 'geometry', 'directions_at_0',
       'directions_at_1', 'directions_at_2', 'directions_at_3',
       'directions_at_4', 'directions_at_5', 'directions_at_6',
       'directions_at_7', 'directions_at_8', 'directions_at_9',
       'directions_at_10', 'directions_at_11', 'directions_at_12',
       'directions_at_13', 'directions_at_14', 'directions_at_15',
       'directions_at_16', 'directions_at_17', 'directions_at_18',
       'directions_at_19', 'directions_at_20', 'directions_at_21',
       'directions_at_22', 'directions_at_23'],
      dtype='object')

In [5]:

trips_gdf

trips_gdf

Out[5]:

	trips_at_0	trips_at_1	trips_at_2	trips_at_3	trips_at_4	trips_at_5	trips_at_6	trips_at_7	trips_at_8	trips_at_9	...	directions_at_14	directions_at_15	directions_at_16	directions_at_17	directions_at_18	directions_at_19	directions_at_20	directions_at_21	directions_at_22	directions_at_23
100	0	0	0	0	2	3	4	3	3	3	...	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}	{LEŚNICA}
1000	1	1	1	1	1	5	7	6	9	9	...	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	{DWORZEC GŁÓWNY, KRZYKI}	NaN
1002	2	2	2	1	6	7	9	6	8	5	...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Żórawi...	{Zabrodzie - pętla, DWORZEC AUTOBUSOWY, Wojszy...	{Żórawina - Niepodległości (Mostek), Wojszycka...	{GAJ - pętla, Wojszycka, JARNOŁTÓW, KRZYKI}
1004	4	3	3	3	11	18	21	12	9	6	...	{PRACZE ODRZAŃSKIE, OPORÓW, Janówek, GAJ - pęt...	{GAJ - pętla, PRACZE ODRZAŃSKIE, KUŹNIKI, OPORÓW}	{GAJ - pętla, KUŹNIKI, OPORÓW}	{GAJ - pętla, KUŹNIKI, OPORÓW}	{GAJ - pętla, KUŹNIKI, OPORÓW}	{GAJ - pętla, KUŹNIKI, RĘDZIŃSKA, OPORÓW}	{GAJ - pętla, KUŹNIKI, OPORÓW}	{OPORÓW, Janówek, Dworzec Główny (Stawowa), GA...	{PRACZE ODRZAŃSKIE, OPORÓW, LEŚNICA, GAJ - pęt...	{GAJ - pętla, KUŹNIKI, OPORÓW}
101	1	0	0	0	0	2	4	4	3	3	...	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska, Zajezdnia Obornicka}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska, Zajezdnia Obornicka}
...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...
993	2	1	1	1	4	8	10	11	10	8	...	{Metalowców, PL. JANA PAWŁA II}	{Metalowców, PL. JANA PAWŁA II}	{Metalowców, PL. JANA PAWŁA II}	{Metalowców, PL. JANA PAWŁA II, Zajezdnia Obor...	{Metalowców, PL. JANA PAWŁA II, Zajezdnia Obor...	{Kwiska, PL. JANA PAWŁA II, Zajezdnia Obornick...	{Kwiska, PL. JANA PAWŁA II, Zajezdnia Obornick...	{PL. JANA PAWŁA II, Zajezdnia Obornicka, Metal...	{PL. JANA PAWŁA II, Zajezdnia Obornicka, Metal...	{Zajezdnia Obornicka, DWORZEC NADODRZE}
994	1	0	0	0	0	2	4	4	3	3	...	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska, Zajezdnia Obornicka}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska, Zajezdnia Obornicka}
997	0	0	0	0	0	5	6	7	6	5	...	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}	{GALERIA DOMINIKAŃSKA}
998	0	0	0	0	0	1	2	1	2	0	...	{KLECINA (Stacja kolejowa)}	{KLECINA (Stacja kolejowa)}	{KLECINA (Stacja kolejowa)}	{KLECINA (Stacja kolejowa), Zajezdnia Obornicka}	{Zajezdnia Obornicka}	NaN	NaN	NaN	NaN	NaN
999	1	1	1	1	1	0	0	0	0	0	...	NaN	NaN	NaN	NaN	NaN	NaN	NaN	NaN	NaN	{KSIĘŻE WIELKIE}

2270 rows × 49 columns