9.9. Getting SQL Data into a DataFrame¶
Running the kind of queries we did in the last few sections is great for exploring and getting to know the data. It is also useful if you just need a one time quick answer. But more often than not what you want to do is to get the data from a database into a DataFrame so that you can visualize it, combine it with other data, or do further analysis. For this exercise our goal is to create a Google Map with a marker for all of the stations in the database.
Pandas provides us with a great way to use our SQL knowledge to create a DataFrame with the results. The
pd.read_sql allows us to pass in a SQL query as a string, along with a URL to connect to the database, and will return a DataFrame. Let’s try that on our bikeshare_stations table now.
import pandas as pd stations = pd.read_sql_query("""select * from bikeshare_stations where latitude is not NULL""",'sqlite:///_static/bikeshare.db') stations.head()
|0||0||31620||5th & F St NW||open||38.897637||-77.018126|
|1||1||31105||14th & Harvard St NW||open||38.926638||-77.032126|
|2||2||31400||Georgia & New Hampshire Ave NW||closed||38.935638||-77.024126|
|3||3||31111||10th & U St NW||open||38.917638||-77.025126|
|4||4||31104||Adams Mill & Columbia Rd NW||open||38.922638||-77.042126|
We can use any SQL query we want and pandas is ‘smart’ enough to look at the results and create columns that match the columns in the database. If we know we are going to have columns containing dates its a good idea to use the
parse_dates parameter when we call
read_sql_query This will ensure that our date columns are of the correct type without us having to convert them later.
Create a DataFrame of the first 10 rides from the trip_data table sorted by start_date. Then add a column called
calc_duration where you calculate the duration of the ride by subtracting the start date from the end date.
9.9.1. Mapping Bike Stations Using Anaconda¶
This section specifically references Jupyterlab and uses a module not supported by Google Colab. If you are using Colab, proceed to the next section where we will learn about how to map bike stations using Colab.
Now let’s have some fun. We are going to map the location of all of the bike stations on a google map! Later you can explore all kinds of ways to visualize rides and ride frequencies routes between starting and ending point and many more things.
To map our stations we will use the
ipyleaflet module. You will need to
conda install -c conda-forge ipyleaflet to install this. You will also need to install the jupyter-leaflet extension for Jupyterlab. You can also enable the extension manager in Jupyterlab and use that to install the extension.
Let’s proceed under the assumption that you have all of the prerequisites installed and working. Our next task is to create a list of all the latitude and longitude values for each of the bikeshare stations. The challenge here is that the Google Maps interface wants to get a list of that looks like this:
[(lat1, long1), (lat2, long2), ...] that means we need to combine the two columns from the DataFrame into a list of tuples. You could, of course, do this with a for loop, iterating over all the rows and making a tuple, but there is a much easier way.
Suppose we have two lists:
list1 = ['a', 'b', 'c'] list2 = [1, 2, 3]
We want to combine these into
[('a', 1), ('b', 2), ('c', 3)] We can do that using Python’s
zip function like
zip(list1, list2). The zip function takes any number of iterable objects and “zips” them together taking the first element from each and making a tuple, then the second element of each etc. If we expand our example
list1 = ['a', 'b', 'c'] list2 = [1, 2, 3] str1 = "XYZ"
list(zip(list1, list2, str1)) returns
[('a', 1, 'X'), ('b', 2, 'Y'), ('c', 3, 'Z')] But, what is the deal with wrapping
zip does not actually return a list, zip returns an iterator that in most circumstances behaves just like a list, except when you want to see a small example.
The good news is that series are also iterables so we can pass a series to the zip function and it will work great.
import pandas as pd stations = pd.read_csv("https://runestone.academy/ns/books/published/httlads/_static/bikeshare_stations.csv") stations.head()
|0||31620||5th & F St NW||open||38.896301||-77.027463|
|1||31105||14th & Harvard St NW||open||38.926571||-77.032414|
|2||31400||Georgia & New Hampshire Ave NW||open||39.221500||-77.059858|
|3||31111||10th & U St NW||open||38.916804||-77.026235|
|4||31104||Adams Mill & Columbia Rd NW||open||38.922680||-77.042667|
First, we can load the stations data. You can get this from your database, from a local file, or even remotely as we are demonstrating here. Next we will use the latitude and longitude data to create a marker for each station.
from ipyleaflet import Map, Marker, Icon, CircleMarker locations = list(zip(stations.latitude, stations.longitude)) dc_center = (38.9072, -77.0369) dcmap = Map(center=dc_center, zoom=12) for loc in locations: marker = CircleMarker(location=loc, radius=2) dcmap.add_layer(marker) dcmap
9.9.2. Practice and Further Exploration¶
The map seems a little busy with all those markers. Redo the map so that it only shows the 20 busiest stations. That is the 20 stations with the highest starting counts.
It would also be interesting to add the 20 stations with the largest ending counts. Can you add those in a different color?
Investigate the interface to see if there is a way for you to color code the markers based on the number of rides originating from that station. Show the 50 most popular stations using five different colors. 1 color for the top 10 another for the next 10 and so on.
Bikeshare datasets are available for many cities. Most of them come in a similar format to this one. Find some bikeshare data for a city close to you or for your favorite city and see if you can reproduce this map. Hint, if your data does not come with latitude and longitude then investigate the
geopypackage, you can use a free service like GeocodeFarm to use the address of the station to get the latitude and longitude.
Here is a real challenge for you, pick a station and then follow the rentals, but only map the stations where a bike ends up back at your original starting point. In other words for the bikes that started at station A, go to station B, then D then pause for a while then on to station X and finally back to A. Others may take a different route. If you can do this you can investigate the AntPath layer that let’s you add a route to the map! Then you can show the routes that the various bikes took to make their way back to the starting point. You may want to limit the time on this to one day or a week.
Another Challenge The
ipyleafletinterface allows you to add a heatmap layer. This sounds like it could a very interesting way to overlay the popularity of different stations and routes on the map. Investigate this layer and what the data should look like, then see if you can find a way to get the data into the appropriate form to make a heatmap.
To get started you can download
pandas_and_SQL.ipynb and open it in Jupyterlab.