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, 20.7MiB/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	...	{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, DWORZEC GŁÓWNY}	NaN
1002	2	2	2	1	6	7	9	6	8	5	...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, Zabrodzie - pętla, Żórawi...	{DWORZEC AUTOBUSOWY, DWORZEC GŁÓWNY, Wojszycka...	{DWORZEC AUTOBUSOWY, Wojszycka, Żórawina - Nie...	{KRZYKI, JARNOŁTÓW, Wojszycka, GAJ - pętla}
1004	4	3	3	3	11	18	21	12	9	6	...	{PRACZE ODRZAŃSKIE, OPORÓW, GAJ - pętla, KUŹNI...	{OPORÓW, GAJ - pętla, KUŹNIKI, PRACZE ODRZAŃSKIE}	{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}	{Dworzec Główny (Stawowa), OPORÓW, GAJ - pętla...	{PRACZE ODRZAŃSKIE, OPORÓW, GAJ - pętla, KUŹNI...	{OPORÓW, GAJ - pętla, KUŹNIKI}
101	1	0	0	0	0	2	4	4	3	3	...	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska}	{Kwiska, Zajezdnia Obornicka}	{Kwiska}	{Kwiska}	{Kwiska}	{Zajezdnia Obornicka, Kwiska}
...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...	...
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}	{Zajezdnia Obornicka, Metalowców, PL. JANA PAW...	{Zajezdnia Obornicka, Metalowców, PL. JANA PAW...	{Zajezdnia Obornicka, Metalowców, Kwiska, PL. ...	{Kwiska, Metalowców, Zajezdnia Obornicka, PL. ...	{Zajezdnia Obornicka, Metalowców, PL. JANA PAW...	{Zajezdnia Obornicka, Metalowców, PL. JANA PAW...	{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}	{Zajezdnia Obornicka, Kwiska}
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)}	{Zajezdnia Obornicka, KLECINA (Stacja kolejowa)}	{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