Code in support of this post: Analyzing 1.1 Billion NYC Taxi and Uber Trips, with a Vengeance
This repo provides scripts to download, process, and analyze data for over 1.1 billion taxi and Uber trips originating in New York City. The data is stored in a PostgreSQL database, and uses PostGIS for spatial calculations, in particular mapping latitude/longitude coordinates to census tracts.
The yellow and green taxi data comes from the NYC Taxi & Limousine Commission, and Uber data comes via FiveThirtyEight, who obtained it via a FOIL request.
Your mileage may vary, but on my MacBook Air, this process took about 3 days to complete. The unindexed database takes up 267 GB on disk. Adding indexes for improved query performance increases total disk usage to 375 GB.
1. Install PostgreSQL and PostGIS
Both are available via Homebrew on Mac OS X
./download_raw_data.sh
./initialize_database.sh
./import_trip_data.sh
./download_raw_uber_data.sh
./import_uber_trip_data.sh
Additional Postgres and R scripts for analysis are in the analysis/
folder, or you can do your own!
trips
table contains all yellow and green taxi trips, plus Uber pickups from April 2014 through September 2014. Each trip has acab_type_id
, which references thecab_types
table and refers to one ofyellow
,green
, oruber
. Each trip maps to a census tract for pickup and dropoffnyct2010
table contains NYC census tracts, plus a fake census tract for the Newark Airport. It also maps census tracts to NYC's official neighborhood tabulation areasuber_trips_2015
table contains Uber pickups from January 2015 through June, 2015. These are kept in a separate table because they don't have specific latitude/longitude coordinates, only location IDs. The location IDs are stored in theuber_taxi_zone_lookups
table, which also maps them (approximately) to neighborhood tabulation areascentral_park_weather_observations
has summary weather data by date
These are bundled with the repository, so no need to download separately, but:
- Shapefile for NYC census tracts and neighborhood tabulation areas comes from Bytes of the Big Apple
- Central Park weather data comes from the National Climatic Data Center
- Remove carriage returns and empty lines from TLC data before passing to Postgres
COPY
command green
taxi raw data files have extra columns with empty data, had to create dummy columnsjunk1
andjunk2
to absorb them- Two of the
yellow
taxi raw data files had a small number of rows containing extra columns. I discarded these rows - The official NYC neighborhood tabulation areas (NTAs) included in the shapefile are not exactly what I would have expected. Some of them are bizarrely large and contain more than one neighborhood, e.g. "Hudson Yards-Chelsea-Flat Iron-Union Square", while others are confusingly named, e.g. "North Site-South Side" for what I'd call "Williamsburg", and "Williamsburg" for what I'd call "South Williamsburg". In a few instances I modified NTA names, but I kept the NTA geographic definitions
- The shapefile includes only NYC census tracts. Trips to New Jersey, Long Island, Westchester, and Connecticut are not mapped to census tracts, with the exception of the Newark Airport, for which I manually added a fake census tract
- The Uber 2015 data uses location IDs instead of latitude/longitude. The location IDs do not exactly overlap with the NYC neighborhood tabulation areas (NTAs) or census tracts, but I did my best to map Uber location IDs to NYC NTAs
Google BigQuery and Amazon Redshift would probably provide significant performance improvements over PostgreSQL. A lot of the data is already available on BigQuery, but in scattered tables, and each trip has only by latitude and longitude coordinates, not census tracts and neighborhoods. PostGIS seemed like the easiest way to map coordinates to census tracts. Once the mapping is complete, it might make sense to load the data back into BigQuery or Redshift to make the analysis faster. Note that BigQuery and Redshift cost some amount of money, while PostgreSQL and PostGIS are free.
There's a Ruby script in the tlc_statistics/
folder to import data from the TLC's summary statistics reports:
ruby import_statistics_data.rb
[email protected], or open a GitHub issue