diff --git a/gtfs_functions/gtfs_functions.py b/gtfs_functions/gtfs_functions.py index a47867f..371af41 100644 --- a/gtfs_functions/gtfs_functions.py +++ b/gtfs_functions/gtfs_functions.py @@ -622,7 +622,7 @@ def get_trips(self): # To allow processing incomplete GTFS data, we must reindex instead: # https://pandas.pydata.org/pandas-docs/stable/user_guide/indexing.html#deprecate-loc-reindex-listlike # This will add NaN for any missing columns. - cols = ['trip_id', 'route_id', 'route_name', 'service_id', 'direction_id', 'shape_id'] + cols = ['trip_id', 'trip_headsign', 'route_id', 'route_name', 'service_id', 'direction_id', 'shape_id'] trips = add_route_name(trips, routes).reindex(columns=cols) # Fill null values