Learning Objectives
By the end of this lesson the learner will know how to:
- Access a database from R
- Run SQL queries in R using RSQLite and dplyr.
- Create an SQLite database from existing .csv files.
So far, we have dealt with small datasets that easily fit into your computer’s memory. But what about datasets that are too large for your computer to handle as a whole? In this case, storing the data outside of R and organizing it in a database is helpful: connecting to the database allows you to retrieve only the chunks needed for the current analysis.
Even better, many large datasets are already available in public or private databases. You can query them without having to download the data first.
R can connect to almost any existing database type. Most common database types have R packages that allow you to connect to them (e.g., RSQLite, RMySQL, etc). Futhermore, the dplyr package you used in the previous chapter supports connecting to the widely-used open source databases sqlite, mysql and postgresql, as well as Google’s bigquery, and it can also be extended to other database types (a vignette in the dplyr package explains how to do it).
Interfacing with databases using dplyr
focusses on retrieving and analyzing datasets by generating SELECT
SQL statements, but it doesn’t modify the database itself. dplyr
does not offer functions to UPDATE
or DELETE
entries. If you need these functionalities, you will need to use the dedicated R packages (e.g., RSQLite). Here we will demonstrate how to interact with a database using dplyr, using both the dplyr’s verb syntax and the SQL syntax.
We will continue to explore the surveys
data you are already familiar with from previous lessons.
The SQLite database is contained in a single file portal_mammals.sqlite
that you generated during the SQL lesson. If you don’t have it, you can download it from Figshare into the data
subdirectory using:
dir.create("data", showWarnings = FALSE)
download.file(url = "https://ndownloader.figshare.com/files/2292171",
destfile = "data/portal_mammals.sqlite")
We can point R to this database with dplyr's
src_sqlite()
command.
library(dplyr)
mammals <- src_sqlite("data/portal_mammals.sqlite")
The src_sqlite()
command does not load the data into the R session (as the read.csv()
function did). Instead, it merely instructs R to connect to the SQLite
database contained in the portal_mammals.sqlite
file.
(You can use the src_mysql()
, src_postgres()
and src_bigquery()
to connect to the other database types supported by dplyr
.)
Let’s take a closer look at the mammals
database we just connected to
mammals
#> src: sqlite 3.11.1 [data/portal_mammals.sqlite]
#> tbls: plots, species, surveys
Just like a spreadsheet with multiple worksheets, a SQLite database can contain multiple tables. In this case three of them are listed in the tbls
row in the output above:
Now that we know that we can connect to the database, let’s explore how to get the data from its tables into R.
To connect to tables within a database, you can use the tbl()
function from dplyr. This function can be used to send SQL queries to the database. To demonstrate this functionality, let’s select the columns “year”, “species_id”, and “plot_id” from the surveys
table:
tbl(mammals, sql("SELECT year, species_id, plot_id FROM surveys"))
With this approach you can use any of the SQL queries we have seen in the database lesson.
One of the strengths of dplyr, is that the same operation can be done using the dplyr’s verbs instead of writing SQL. First, we select the table on which to do the operations by creating the surveys
object, and then we use the standard dplyr syntax as if it were a data frame:
surveys <- tbl(mammals, "surveys")
surveys %>%
select(year, species_id, plot_id)
In this case, the surveys
object behaves like a data frame, and several of the functions that can used with data frames, can also be used on tables from a database. For instance, the head()
function can be used to check the first 10 rows of the table:
head(surveys, n = 10)
#> Source: query [?? x 9]
#> Database: sqlite 3.11.1 [data/portal_mammals.sqlite]
#>
#> record_id month day year plot_id species_id sex hindfoot_length
#> <int> <int> <int> <int> <int> <chr> <chr> <int>
#> 1 1 7 16 1977 2 NL M 32
#> 2 2 7 16 1977 3 NL M 33
#> 3 3 7 16 1977 2 DM F 37
#> 4 4 7 16 1977 7 DM M 36
#> 5 5 7 16 1977 3 DM M 35
#> 6 6 7 16 1977 1 PF M 14
#> 7 7 7 16 1977 2 PE F NA
#> 8 8 7 16 1977 1 DM M 37
#> 9 9 7 16 1977 1 DM F 34
#> 10 10 7 16 1977 6 PF F 20
#> # ... with more rows, and 1 more variables: weight <int>
This output of the head
command looks just like a regular data.frame
: The table has 9 columns and the head()
command shows us the first 10 rows. Note that the columns plot_type
, taxa
, genus
, and species
are missing. These are now located in the tables plots
and species
which we will join together in a moment.
However, some functions don’t work quite as expected. For instance, let’s check how many rows there are in total using nrow()
:
nrow(tbl)
#> NULL
That’s strange - R doesn’t know how many rows the survey
table contains - it returns NULL
instead. You might have already noticed that the first line of the head()
output included ??
indicating that the number of rows wasn’t known.
The reason for this behavior highlights a key difference between using dplyr
on datasets in memory (e.g. loaded into your R session via read.csv()
) and those provided by a database. To understand it, we take a closer look at how dplyr
communicates with our SQLite database.
Relational databases typically use a special-purpose language, Structured Query Language (SQL), to manage and query data.
For example, the following SQL query returns the first 10 rows from the surveys
table:
SELECT *
FROM `surveys`
LIMIT 10
Behind the scenes, dplyr
:
To lift the curtain, we can use dplyr's
explain()
function to show which SQL commands are actually sent to the database:
explain(head(surveys, n = 10))
#> <SQL>
#> SELECT *
#> FROM `surveys`
#> LIMIT 10
#>
#> <PLAN>
#> addr opcode p1 p2 p3 p4 p5 comment
#> 1 0 Init 0 18 0 00 NA
#> 2 1 Integer 10 1 0 00 NA
#> 3 2 OpenRead 0 2 0 9 00 NA
#> 4 3 Rewind 0 16 0 00 NA
#> 5 4 Column 0 0 2 00 NA
#> 6 5 Column 0 1 3 00 NA
#> 7 6 Column 0 2 4 00 NA
#> 8 7 Column 0 3 5 00 NA
#> 9 8 Column 0 4 6 00 NA
#> 10 9 Column 0 5 7 00 NA
#> 11 10 Column 0 6 8 00 NA
#> 12 11 Column 0 7 9 00 NA
#> [ reached getOption("max.print") -- omitted 9 rows ]
The first part of the output shows the actual SQL query sent to the database; it matches our manually constructed SELECT
statement above.
Instead of having to formulate the SQL query ourselves - and having to mentally switch back and forth between R and SQL syntax - we can delegate this translation to dplyr
. (You don’t even need to know SQL to interact with a database via dplyr
!)
dplyr
, in turn, doesn’t do the real work of subsetting the table, either. Instead, it merely sends the query to the database, waits for its response and returns it to us.
That way, R never gets to see the full surveys
table - and that’s why it could not tell us how many rows it contains. On the bright side, this allows us to work with large datasets - even too large to fit into our computer’s memory.
dplyr
can translate many different query types into SQL allowing us e.g. to select()
specific columns, filter()
rows or join tables.
To see this in action, let’s compose a few queries with dplyr
.
First, let’s only request rows of the surveys
table in which weight
is less than 5 and keep only the species_id, sex, and weight columns.
surveys %>%
filter(weight < 5) %>%
select(species_id, sex, weight)
#> Source: query [?? x 3]
#> Database: sqlite 3.11.1 [data/portal_mammals.sqlite]
#>
#> species_id sex weight
#> <chr> <chr> <int>
#> 1 PF M 4
#> 2 PF F 4
#> 3 PF <NA> 4
#> 4 PF F 4
#> 5 PF F 4
#> 6 RM M 4
#> 7 RM F 4
#> 8 RM M 4
#> 9 RM M 4
#> 10 RM M 4
#> # ... with more rows
Executing this command will return a table with 10 rows and the requested species_id
, sex
and weight
columns. Great!
… but wait, why are there only 10 rows?
The last line:
# ... with more rows
indicates that there are more results that fit our filtering criterion. Why was R lazy and only retrieved 10 of them?
Hadley Wickham, the author of dplyr
explains:
When working with databases, dplyr tries to be as lazy as possible:
- It never pulls data into R unless you explicitly ask for it.
- It delays doing any work until the last possible moment: it collects together everything you want to do and then sends it to the database in one step.
When you construct a dplyr
query, you can connect multiple verbs into a single pipeline. For example, we combined the filter()
and select()
verbs using the %>%
pipe.
If we wanted to, we could add on even more steps, e.g. remove the sex
column in an additional select
call:
data_subset <- surveys %>%
filter(weight < 5) %>%
select(species_id, sex, weight)
data_subset %>%
select(-sex)
#> Source: query [?? x 2]
#> Database: sqlite 3.11.1 [data/portal_mammals.sqlite]
#>
#> species_id weight
#> <chr> <int>
#> 1 PF 4
#> 2 PF 4
#> 3 PF 4
#> 4 PF 4
#> 5 PF 4
#> 6 RM 4
#> 7 RM 4
#> 8 RM 4
#> 9 RM 4
#> 10 RM 4
#> # ... with more rows
Just like the first select(species_id, sex, weight)
call, the select(-sex)
command is not executed by R. It is sent to the database instead. Only the final result is retrieved and displayed to you.
Of course, we could always add on more steps, e.g. we could filter by species_id
or minimum weight
. That’s why R doesn’t retrieve the full set of results - instead it only retrieves the first 10 results from the database by default. (After all, you might want to add an additional step and get the database to do more work…)
To instruct R to stop being lazy, e.g. to retrieve all of the query results from the database, we add the collect()
command to our pipe. It indicates that our database query is finished: time to get the final results and load them into the R session.
data_subset <- surveys %>%
filter(weight < 5) %>%
select(species_id, sex, weight) %>%
collect()
Now we have all 17 rows that match our query in a data.frame
and can continue to work with them exclusively in R, without communicating with the database.
dplyr
enables database queries across one or multiple database tables, using the same single- and multiple-table verbs you encountered previously. This means you can use the same commands regardless of whether you interact with a remote database or local dataset! This is a really useful feature if you work with large datasets: you can first prototype your code on a small subset that fits into memory, and when your code is ready, you can change the input dataset to your full database without having to change the syntax.
On the other hand, being able use SQL queries directly can be useful if your collaborators have already put together complex queries to prepare the dataset that you need for your analysis.
To illustrate how to use dplyr with these complex queries, we are going to join the plots
and surveys
tables. The plots
table in the database contains information about the different plots surveyed by the reseachers. To access it, we point the tbl()
command to it:
plots <- tbl(mammals, "plots")
plots
#> Source: query [?? x 2]
#> Database: sqlite 3.11.1 [data/portal_mammals.sqlite]
#>
#> plot_id plot_type
#> <int> <chr>
#> 1 1 Spectab exclosure
#> 2 2 Control
#> 3 3 Long-term Krat Exclosure
#> 4 4 Control
#> 5 5 Rodent Exclosure
#> 6 6 Short-term Krat Exclosure
#> 7 7 Rodent Exclosure
#> 8 8 Control
#> 9 9 Spectab exclosure
#> 10 10 Rodent Exclosure
#> # ... with more rows
The plot_id
column also features in the surveys
table.
surveys
#> Source: query [?? x 9]
#> Database: sqlite 3.11.1 [data/portal_mammals.sqlite]
#>
#> record_id month day year plot_id species_id sex hindfoot_length
#> <int> <int> <int> <int> <int> <chr> <chr> <int>
#> 1 1 7 16 1977 2 NL M 32
#> 2 2 7 16 1977 3 NL M 33
#> 3 3 7 16 1977 2 DM F 37
#> 4 4 7 16 1977 7 DM M 36
#> 5 5 7 16 1977 3 DM M 35
#> 6 6 7 16 1977 1 PF M 14
#> 7 7 7 16 1977 2 PE F NA
#> 8 8 7 16 1977 1 DM M 37
#> 9 9 7 16 1977 1 DM F 34
#> 10 10 7 16 1977 6 PF F 20
#> # ... with more rows, and 1 more variables: weight <int>
Because plot_id
is listed in both tables, we can use it to look up matching records, and join the two tables. For example, to extract all surveys for the first plot, which has plot_id
1, we can do:
plots %>%
filter(plot_id == 1) %>%
inner_join(surveys) %>%
collect()
#> # A tibble: 1,995 × 10
#> plot_id plot_type record_id month day year species_id sex
#> <int> <chr> <int> <int> <int> <int> <chr> <chr>
#> 1 1 Spectab exclosure 6 7 16 1977 PF M
#> 2 1 Spectab exclosure 8 7 16 1977 DM M
#> 3 1 Spectab exclosure 9 7 16 1977 DM F
#> 4 1 Spectab exclosure 78 8 19 1977 PF M
#> 5 1 Spectab exclosure 80 8 19 1977 DS M
#> 6 1 Spectab exclosure 218 9 13 1977 PF M
#> 7 1 Spectab exclosure 222 9 13 1977 DS M
#> 8 1 Spectab exclosure 239 9 13 1977 DS M
#> 9 1 Spectab exclosure 263 10 16 1977 DM M
#> 10 1 Spectab exclosure 270 10 16 1977 DM F
#> # ... with 1,985 more rows, and 2 more variables: hindfoot_length <int>,
#> # weight <int>
Important Note: Without the collect()
statement, only the first 10 matching rows are returned. By adding collect()
, the full set of 1,985 is retrieved.
Challenge
Write a query that returns the number of rodents observed in each plot in each year.
Hint: Connect to the species table and write a query that joins the species and survey tables together to exclude all non-rodents. The query should return counts of rodents by year.
Optional: Write a query in SQL that will produce the same result. You can join multiple tables together uusing the following syntax where foreign key refers to your unique id (e.g.,
species_id
):SELECT table.col, table.col FROM table1 JOIN table2 ON table1.key = table2.key JOIN table3 ON table2.key = table3.key
Challenge
Write a query that returns the total number of rodents in each genus caught in the different plot types.
Hint: Write a query that joins the species, plot, and survey tables together. The query should return counts of genus by plot type.
This is useful if we are interested in estimating the number of individuals belonging to each genus found in each plot type. But what if we were interested in the number of genera found in each plot type? Using tally()
gives the number of individuals, instead we need to use n_distinct()
to count the number of unique values found in a column.
unique_genera <- left_join(surveys, plots) %>%
left_join(species) %>%
group_by(plot_type) %>%
summarize(
n_genera = n_distinct(genus)
) %>%
collect()
n_distinct
, like the other dplyr functions we have used in this lesson, works not only on database connections but also on regular data frames.
So far, we have used a previously prepared SQLite database. But we can also use R to create a new database, e.g. from existing csv
files. Let’s recreate the mammals database that we’ve been working with, in R. First let’s read in the csv
files.
species <- read.csv("data/species.csv")
surveys <- read.csv("data/surveys.csv")
plots <- read.csv("data/plots.csv")
Creating a new SQLite database with dplyr
is easy. You can re-use the same command we used above to open an existing .sqlite
file. The create = TRUE
argument instructs R to create a new, empty database instead.
Caution: When create = TRUE
is added, any existing database at the same location is overwritten without warning.
myDB <- "portalR.sqlite"
my_db <- src_sqlite(myDB, create = TRUE)
Currently, our new databse is empty, it doesn’t contain any tables:
my_db
#> src: sqlite 3.11.1 [portalR.sqlite]
#> tbls:
To add tables, we copy the existing data.frames into the database one by one:
copy_to(my_db, surveys)
copy_to(my_db, plots)
my_db
If you check the location of our database you’ll see that data is automatically being written to disk. R and dplyr
not only provide easy ways to query existing databases, they also allows you to easily create your own databases from flat files!
Challenge
Add the remaining species table to the my_db database and run some of your queries from earlier in the lesson to verify that you have faithfully recreated the mammals database.
Note: In this example, we first loaded all of the data into the R session by reading the three csv files. Because all the data has to flow through R, this is not suitable for very large datasets.
Data Carpentry, 2017.
License. Questions? Feedback?
Please file
an issue on GitHub.
On Twitter: @datacarpentry