8. Merge

Next we’ll cover how to merge two DataFrames together into a combined table.

We’ll pull faa-survey.csv, which contains annual estimates of how many hours each type of helicopter was in the air. If we merge it with our accident totals, we will be able to calculate an accident rate.

We can read it in the same way as the NTSB accident list, with read_csv.

Hide code cell content
import pandas as pd
accident_list = pd.read_csv("https://raw.githubusercontent.com/palewire/first-python-notebook/main/docs/src/_static/ntsb-accidents.csv")
accident_list["latimes_make_and_model"] = accident_list["latimes_make_and_model"].str.upper()
accident_counts = accident_list.groupby("latimes_make_and_model").size().reset_index().rename(columns={0: "accidents"})
survey = pd.read_csv("https://raw.githubusercontent.com/palewire/first-python-notebook/main/docs/src/_static/faa-survey.csv")

When joining two tables together, the first step is to look carefully at the columns in each table to find one they have in common and can serve as the basis for the join. We can do that with the info command we learned earlier.

accident_counts.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 12 entries, 0 to 11
Data columns (total 2 columns):
 #   Column                  Non-Null Count  Dtype 
---  ------                  --------------  ----- 
 0   latimes_make_and_model  12 non-null     object
 1   accidents               12 non-null     int64 
dtypes: int64(1), object(1)
memory usage: 324.0+ bytes
survey.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 12 entries, 0 to 11
Data columns (total 2 columns):
 #   Column                  Non-Null Count  Dtype 
---  ------                  --------------  ----- 
 0   latimes_make_and_model  12 non-null     object
 1   total_hours             12 non-null     int64 
dtypes: int64(1), object(1)
memory usage: 324.0+ bytes

You can see that each table contains the latimes_make_and_model column. We can therefore join the two files using that column with the pandas merge method.

Note

If you are familar with traditional databases, you may recognize that the merge method in pandas is similar to SQL’s JOIN statement. If you dig into merge’s documentation you will see it has many of the same options.

Merging two DataFrames is as simple as passing both to pandas’ built-in merge method and specifying which field you’d like to use to connect them. We will save the result into another new variable, which I’m going to call merged_list.

merged_list = pd.merge(accident_counts, survey, on="latimes_make_and_model")

Note

You may notice something new with the on="latimes_make_and_model" bit above. It is what Python calls a keyword argument. Keyword arguments are inputs passed to a function or method after explicitly specifying the name of the argument, followed by an equal sign.

Keyword arguments can be passed in any order, as long as the name of the argument is specified. When creating a function, they can be used to specify a default value for a parameter. For this reason, they are commonly used to provide overrides of a method’s out-of-the-box behavior.

The pandas documentation for merge reveals all of the keyword options available, as well as their defaults.

That new DataFrame can be inspected like any other.

merged_list.head()
latimes_make_and_model accidents total_hours

Gasp! There’s nothing there! What happened? Let’s go back and inspect the datasets we’re trying to merge. The head command remains our trusty friend for this type of task.

First, there was the accident counts.

accident_counts.head()
latimes_make_and_model accidents
0 AGUSTA 109 2
1 AIRBUS 130 1
2 AIRBUS 135 4
3 AIRBUS 350 29
4 BELL 206 30

Then, there was the FAA survey dataset.

survey.head()
latimes_make_and_model total_hours
0 AGuSTA 109 362172
1 airbus 130 1053786
2 AIrBuS 135 884596
3 Airbus 350 3883490
4 bELL 206 5501308

It looks like even though the latimes_make_and_model column represents the same data in each dataset, the casing is messy in the FAA survey data. Raw data is usually messy (even if this particular example is contrived). It’s always important to inspect your data thoroughly and know how to clean it up before analyzing.

Since the uppercase accident counts data is more consistent, let’s modify the FAA data to match. There are a handful of ways to do this, but the most straightforward is simply replacing everything in the latimes_make_and_model column with an uppercase copy of itself, as we did with the accident data in an earlier chapter. Once again, the str method can do the job.

survey["latimes_make_and_model"] = survey["latimes_make_and_model"].str.upper()

Now let’s try merging our data again.

merged_list = pd.merge(accident_counts, survey, on="latimes_make_and_model")

And then take a peek.

merged_list.head()
latimes_make_and_model accidents total_hours
0 AGUSTA 109 2 362172
1 AIRBUS 130 1 1053786
2 AIRBUS 135 4 884596
3 AIRBUS 350 29 3883490
4 BELL 206 30 5501308

Much better! By looking at the columns, you can check how many rows survived the merge, a precaution you should take every time you join two tables.

merged_list.info()
<class 'pandas.core.frame.DataFrame'>
RangeIndex: 12 entries, 0 to 11
Data columns (total 3 columns):
 #   Column                  Non-Null Count  Dtype 
---  ------                  --------------  ----- 
 0   latimes_make_and_model  12 non-null     object
 1   accidents               12 non-null     int64 
 2   total_hours             12 non-null     int64 
dtypes: int64(2), object(1)
memory usage: 420.0+ bytes

You can also verify that the DataFrame has the same number of records as there are values in accident_totals column. That’s good; if there are no null values, that means that every record in each DataFrame found a match in the other.

Another simple approach is to simply ask pandas to print the number of rows in the DataFrame. That can be done with the len function.

len(merged_list)
12

That number should match the number of rows in the accident totals DataFrame.

len(accident_counts)
12

Now that we are confident we have a properly merged DataFrame, we’re ready to move on to the next step: calculating the accident rate.