r4ds/databases.qmd

681 lines
28 KiB
Plaintext
Raw Normal View History

# Databases {#sec-import-databases}
2021-02-22 20:28:26 +08:00
```{r}
#| results: "asis"
#| echo: false
source("_common.R")
2022-06-04 22:40:20 +08:00
status("polishing")
2021-05-04 21:10:39 +08:00
```
2022-05-11 06:26:04 +08:00
## Introduction
2022-06-02 23:33:07 +08:00
A huge amount of data lives in databases, so it's essential that you know how to access it.
Sometimes you can ask someone to download a snapshot into a .csv for you, but this gets painful quickly: every time you need to make a change you'll have to communicate with another human.
2022-05-14 02:38:55 +08:00
You want to be able to reach into the database directly to get the data you need, when you need it.
2022-08-04 21:00:38 +08:00
In this chapter, you'll first learn the basics of the DBI package: how to use it to connect to a database and then retrieve data with a SQL[^databases-1] query.
2022-06-02 23:33:07 +08:00
**SQL**, short for **s**tructured **q**uery **l**anguage, is the lingua franca of databases, and is an important language for all data scientists to learn.
That said, we're not going to start with SQL, but instead we'll teach you dbplyr, which can translate your dplyr code to the SQL.
2022-05-24 23:12:41 +08:00
We'll use that as way to teach you some of the most important features of SQL.
You won't become a SQL master by the end of the chapter, but you will be able to identify the most important components and understand what they do.
2022-05-14 02:38:55 +08:00
2022-08-04 21:00:38 +08:00
[^databases-1]: SQL is either pronounced "s"-"q"-"l" or "sequel".
2022-05-11 06:26:04 +08:00
### Prerequisites
2022-06-02 23:33:07 +08:00
In this chapter, we'll introduce DBI and dbplyr.
DBI is a low-level interface that connects to databases and executes SQL; dbplyr is a high-level interface that translates your dplyr code to SQL queries then executes them with DBI.
2022-05-24 23:12:41 +08:00
```{r}
#| label: setup
#| message: false
2022-05-11 06:26:04 +08:00
library(DBI)
2022-05-24 23:12:41 +08:00
library(dbplyr)
2022-05-11 06:26:04 +08:00
library(tidyverse)
```
## Database basics
2022-05-24 23:12:41 +08:00
At the simplest level, you can think about a database as a collection of data frames, called **tables** in database terminology.
2022-05-11 06:26:04 +08:00
Like a data.frame, a database table is a collection of named columns, where every value in the column is the same type.
2022-05-24 23:12:41 +08:00
There are three high level differences between data frames and database tables:
2022-05-11 06:26:04 +08:00
- Database tables are stored on disk and can be arbitrarily large.
2022-06-02 23:33:07 +08:00
Data frames are stored in memory, and are fundamentally limited (although that limit is still plenty large for many problems).
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
- Database tables almost always have indexes.
Much like the index of a book, a database index makes it possible to quickly find rows of interest without having to look at every single row.
2022-05-11 06:26:04 +08:00
Data frames and tibbles don't have indexes, but data.tables do, which is one of the reasons that they're so fast.
2022-05-24 23:12:41 +08:00
- Most classical databases are optimized for rapidly collecting data, not analyzing existing data.
These databases are called **row-oriented** because the data is stored row-by-row, rather than column-by-column like R.
More recently, there's been much development of **column-oriented** databases that make analyzing the existing data much faster.
2022-06-02 23:33:07 +08:00
Databases are run by database management systems (**DBMS**'s for short), which come in three basic forms:
- **Client-server** DBMS's run on a powerful central server, which you connect from your computer (the client). They are great for sharing data with multiple people in an organisation. Popular client-server DBMS's include PostgreSQL, MariaDB, SQL Server, and Oracle.
- **Cloud** DBMS's, like Snowflake, Amazon's RedShift, and Google's BigQuery, are similar to client server DBMS's, but they run in the cloud. This means that they can easily handle extremely large datasets and can automatically provide more compute resources as needed.
- **In-process** DBMS's, like SQLite or duckdb, run entirely on your computer. They're great for working with large datasets where you're the primary user.
2022-05-11 06:26:04 +08:00
## Connecting to a database
2022-05-24 23:12:41 +08:00
To connect to the database from R, you'll use a pair of packages:
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
- You'll always use DBI (**d**ata**b**ase **i**nterface) because it provides a set of generic functions that connect to the database, upload data, run SQL queries, etc.
- You'll also use a package tailored for the DBMS you're connecting to.
This package translates the generic DBI commands into the specifics needed for a given DBMS.
There's usually one package for each DMBS, e.g.
RPostgres for Postgres and RMariaDB for MySQL.
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
If you can't find a specific package for your DBMS, you can usually use the odbc package instead.
This uses the ODBC protocol supported by many DBMS.
odbc requires a little more setup because you'll also need to install an ODBC driver and tell the odbc package where to find it.
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
Concretely, you create a database connection using `DBI::dbConnect()`.
2022-08-04 21:00:38 +08:00
The first argument selects the DBMS[^databases-2], then the second and subsequent arguments describe how to connect to it (i.e. where it lives and the credentials that you need to access it).
2022-06-02 23:33:07 +08:00
The following code shows a couple of typical examples:
2022-05-24 23:12:41 +08:00
2022-08-04 21:00:38 +08:00
[^databases-2]: Typically, this is the only function you'll use from the client package, so we recommend using `::` to pull out that one function, rather than loading the complete package with `library()`.
2022-05-11 06:26:04 +08:00
```{r}
#| eval: false
2022-05-24 23:12:41 +08:00
con <- DBI::dbConnect(
RMariaDB::MariaDB(),
username = "foo"
)
con <- DBI::dbConnect(
RPostgres::Postgres(),
hostname = "databases.mycompany.com",
port = 1234
)
2022-05-11 06:26:04 +08:00
```
2022-06-02 23:33:07 +08:00
The precise details of the connection varies a lot from DBMS to DBMS so unfortunately we can't cover all the details here.
This means you'll need to do a little research on your own.
2022-05-24 23:12:41 +08:00
Typically you can ask the other data scientists in your team or talk to your DBA (**d**ata**b**ase **a**dministrator).
The initial setup will often take a little fiddling (and maybe some googling) to get right, but you'll generally only need to do it once.
2022-05-11 06:26:04 +08:00
### In this book
2022-06-02 23:33:07 +08:00
Setting up a client-server or cloud DBMS would be a pain for this book, so we'll instead use an in-process DBMS that lives entirely in an R package: duckdb.
2022-05-24 23:12:41 +08:00
Thanks to the magic of DBI, the only difference between using duckdb and any other DBMS is how you'll connect to the database.
This makes it great to teach with because you can easily run this code as well as easily take what you learn and apply it elsewhere.
2022-06-02 23:33:07 +08:00
Connecting to duckdb is particularly simple because the defaults create a temporary database that is deleted when you quit R.
2022-05-24 23:12:41 +08:00
That's great for learning because it guarantees that you'll start from a clean slate every time you restart R:
2022-05-11 06:26:04 +08:00
```{r}
con <- DBI::dbConnect(duckdb::duckdb())
```
2022-06-02 23:33:07 +08:00
duckdb is a high-performance database that's designed very much for the needs of a data scientist.
We use it here because it's very to easy to get started with, but it's also capable of handling gigabytes of data with great speed.
If you want to use duckdb for a real data analysis project, you'll also need to supply the `dbdir` argument to make a persistent database and tell duckdb where to save it.
Assuming you're using a project (Chapter -@sec-workflow-scripts-projects)), it's reasonable to store it in the `duckdb` directory of the current project:
2022-05-11 06:26:04 +08:00
```{r}
#| eval: false
2022-05-11 06:26:04 +08:00
con <- DBI::dbConnect(duckdb::duckdb(), dbdir = "duckdb")
```
2022-05-24 23:12:41 +08:00
### Load some data {#sec-load-data}
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
Since this is a new database, we need to start by adding some data.
Here we'll use add `mpg` and `diamonds` datasets from ggplot2 using `DBI::dbWriteTable()`.
The simplest usage of `dbWriteTable()` needs three arguments: a database connection, the name of the table to create in the database, and a data frame of data.
2022-05-11 06:26:04 +08:00
```{r}
dbWriteTable(con, "mpg", ggplot2::mpg)
dbWriteTable(con, "diamonds", ggplot2::diamonds)
```
2022-08-10 00:43:12 +08:00
If you're using duckdb in a real project, we highly recommend learning about `duckdb_read_csv()` and `duckdb_register_arrow()`.
2022-05-24 23:12:41 +08:00
These give you powerful and performant ways to quickly load data directly into duckdb, without having to first load it in to R.
2022-05-14 02:38:55 +08:00
2022-08-04 21:00:38 +08:00
## DBI basics
2022-05-11 06:26:04 +08:00
2022-05-14 02:38:55 +08:00
Now that we've connected to a database with some data in it, lets perform some basic operations with DBI.
### What's there?
2022-05-24 23:12:41 +08:00
The most important database objects for data scientists are tables.
2022-08-04 21:00:38 +08:00
DBI provides two useful functions to either list all the tables in the database[^databases-3] or to check if a specific table already exists:
2022-05-24 23:12:41 +08:00
2022-08-04 21:00:38 +08:00
[^databases-3]: At least, all the tables that you have permission to see.
2022-05-24 23:12:41 +08:00
2022-05-14 02:38:55 +08:00
```{r}
dbListTables(con)
dbExistsTable(con, "foo")
```
2022-05-11 06:26:04 +08:00
### Extract some data
2022-05-24 23:12:41 +08:00
Once you've determined a table exists, you can retrieve it with `dbReadTable()`:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-24 23:12:41 +08:00
con |>
dbReadTable("diamonds") |>
as_tibble()
2022-05-11 06:26:04 +08:00
```
2022-08-10 00:43:12 +08:00
`dbReadTable()` returns a `data.frame` so we use `as_tibble()` to convert it into a tibble so that it prints nicely.
2022-05-11 06:26:04 +08:00
2022-06-04 22:22:05 +08:00
In real life, it's rare that you'll use `dbReadTable()` because often database tables are too big to fit in memory, and you want bring back only a subset of the rows and columns.
2022-05-11 06:26:04 +08:00
2022-05-24 23:12:41 +08:00
### Run a query {#sec-dbGetQuery}
2022-05-11 06:26:04 +08:00
2022-05-24 23:12:41 +08:00
The way you'll usually retrieve data is with `dbGetQuery()`.
It takes a database connection and some SQL code and returns a data frame:
2022-05-11 06:26:04 +08:00
```{r}
2022-06-04 22:22:05 +08:00
sql <- "
SELECT carat, cut, clarity, color, price
FROM diamonds
WHERE price > 15000
"
as_tibble(dbGetQuery(con, sql))
2022-05-11 06:26:04 +08:00
```
2022-06-02 23:33:07 +08:00
Don't worry if you've never seen SQL before; you'll learn more about it shortly.
But if you read it carefully, you might guess that it selects five columns of the diamonds dataset and all the rows where `price` is greater than 15,000.
2022-05-14 02:38:55 +08:00
You'll need to be a little careful with `dbGetQuery()` since it can potentially return more data than you have memory.
2022-05-24 23:12:41 +08:00
We won't discuss it further here, but if you're dealing with very large datasets it's possible to deal with a "page" of data at a time by using `dbSendQuery()` to get a "result set" which you can page through by calling `dbFetch()` until `dbHasCompleted()` returns `TRUE`.
2022-05-11 06:26:04 +08:00
2022-05-24 23:12:41 +08:00
### Other functions
2022-05-14 02:38:55 +08:00
There are lots of other functions in DBI that you might find useful if you're managing your own data (like `dbWriteTable()` which we used in @sec-load-data), but we're going to skip past them in the interest of staying focused on working with data that already lives in a database.
2022-05-14 02:38:55 +08:00
2022-05-24 23:12:41 +08:00
## dbplyr basics
2022-05-11 06:26:04 +08:00
2022-05-24 23:12:41 +08:00
Now that you've learned the low-level basics for connecting to a database and running a query, we're going to switch it up a bit and learn a bit about dbplyr.
2022-06-02 23:33:07 +08:00
dbplyr is a dplyr **backend**, which means that you keep writing dplyr code but the backend executes it differently.
In this, dbplyr translates to SQL; other backends include [dtplyr](https://dtplyr.tidyverse.org) which translates to [data.table](https://r-datatable.com), and [multidplyr](https://multidplyr.tidyverse.org) which executes your code on multiple cores.
2022-05-14 02:38:55 +08:00
2022-06-04 22:22:05 +08:00
To use dbplyr, you must first use `tbl()` to create an object that represents a database table:
2022-05-11 06:26:04 +08:00
```{r}
diamonds_db <- tbl(con, "diamonds")
diamonds_db
```
2022-06-04 22:22:05 +08:00
::: callout-note
There are two other common way to a database.
First, many corporate databases are very large so need some hierarchy to keep all the tables organised.
In that case you might need to supply a schema, or a catalog and a schema, in order to pick the table you're interested in:
```{r}
#| eval: false
diamonds_db <- tbl(con, in_schema("sales", "diamonds"))
diamonds_db <- tbl(con, in_catalog("north_america", "sales", "diamonds"))
```
Other times you might want to use your own SQL query as a starting point:
```{r}
#| eval: false
diamonds_db <- tbl(con, sql("SELECT * FROM diamonds"))
```
:::
2022-05-26 06:03:37 +08:00
This object is **lazy**; when you use dplyr verbs on it, dplyr doesn't do any work: it just records the sequence of operations that you want to perform and only performs them when needed.
For example, take the following pipeline:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-14 02:38:55 +08:00
big_diamonds_db <- diamonds_db |>
filter(price > 15000) |>
select(carat:clarity, price)
2022-05-26 06:03:37 +08:00
2022-05-14 02:38:55 +08:00
big_diamonds_db
```
2022-06-02 23:33:07 +08:00
You can tell this object represents a database query because it prints the DBMS name at the top, and while it tells you the number of columns, it typically doesn't know the number of rows.
2022-05-26 06:03:37 +08:00
This is because finding the total number of rows usually requires executing the complete query, something we're trying to avoid.
You can see the SQL the dbplyr generates by a dbplyr query by calling `show_query()`:
2022-05-14 02:38:55 +08:00
```{r}
big_diamonds_db |>
2022-05-11 06:26:04 +08:00
show_query()
```
2022-05-26 06:03:37 +08:00
To get all the data back into R, you call `collect()`.
Behind the scenes, this generates the SQL, calls `dbGetQuery()` to get the data, then turns the result into a tibble:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-14 02:38:55 +08:00
big_diamonds <- big_diamonds_db |>
2022-05-11 06:26:04 +08:00
collect()
big_diamonds
```
2022-06-04 22:22:05 +08:00
Typically, you'll use dbplyr to select the data you want from the database, performing basic filtering and aggregation using the translations described below.
Then, once you're ready to analyse the data with functions that are unique to R, you'll `collect()` the data to get an in-memory tibble, and continue your work with pure R code.
2022-05-24 23:12:41 +08:00
## SQL
2022-06-02 23:33:07 +08:00
The rest of the chapter will teach you a little SQL through the lens of dbplyr.
2022-08-10 00:43:12 +08:00
It's a rather non-traditional introduction to SQL but we hope it will get you quickly up to speed with the basics.
2022-05-26 23:55:12 +08:00
Luckily, if you understand dplyr you're in a great place to quickly pick up SQL because so many of the concepts are the same.
2022-06-02 23:33:07 +08:00
We'll explore the relationship between dplyr and SQL using a couple of old friends from the nycflights13 package: `flights` and `planes`.
These dataset are easy to get into our learning database because dbplyr has a function designed for this exact scenario:
2022-05-26 06:03:37 +08:00
```{r}
dbplyr::copy_nycflights13(con)
flights <- tbl(con, "flights")
planes <- tbl(con, "planes")
2022-06-02 23:33:07 +08:00
```
2022-05-26 23:55:12 +08:00
2022-06-02 23:33:07 +08:00
```{r}
#| echo: false
2022-05-26 23:55:12 +08:00
options(dplyr.strict_sql = TRUE)
2022-05-24 23:12:41 +08:00
```
2022-05-14 02:38:55 +08:00
### SQL basics
2022-06-02 23:33:07 +08:00
The top-level components of SQL are called **statements**.
2022-05-26 06:03:37 +08:00
Common statements include `CREATE` for defining new tables, `INSERT` for adding data, and `SELECT` for retrieving data.
2022-06-02 23:33:07 +08:00
We will on focus on `SELECT` statements, also called **queries**, because they are almost exclusively what you'll use as a data scientist.
2022-05-14 02:38:55 +08:00
2022-05-26 06:03:37 +08:00
A query is made up of **clauses**.
2022-08-04 21:00:38 +08:00
There are five important clauses: `SELECT`, `FROM`, `WHERE`, `ORDER BY`, and `GROUP BY`. Every query must have the `SELECT`[^databases-4] and `FROM`[^databases-5] clauses and the simplest query is `SELECT * FROM table`, which selects all columns from the specified table
. This is what dplyr generates for an unadulterated table
2022-05-31 21:40:22 +08:00
:
2022-05-14 02:38:55 +08:00
2022-08-04 21:00:38 +08:00
[^databases-4]: Confusingly, depending on the context, `SELECT` is either a statement or a clause.
2022-05-31 21:40:22 +08:00
To avoid this confusion, we'll generally use query instead of `SELECT` statement.
2022-08-04 21:00:38 +08:00
[^databases-5]: Ok, technically, only the `SELECT` is required, since you can write queries like `SELECT 1+1` to perform basic calculations.
2022-05-14 02:38:55 +08:00
But if you want to work with data (as you always do!) you'll also need a `FROM` clause.
2022-05-24 23:12:41 +08:00
```{r}
2022-05-26 06:03:37 +08:00
flights |> show_query()
planes |> show_query()
2022-05-24 23:12:41 +08:00
```
2022-06-02 23:33:07 +08:00
`WHERE` and `ORDER BY` control which rows are included and how they are ordered:
2022-05-11 06:26:04 +08:00
2022-05-26 06:03:37 +08:00
```{r}
flights |>
filter(dest == "IAH") |>
2022-05-31 21:40:22 +08:00
arrange(dep_delay) |>
2022-05-26 06:03:37 +08:00
show_query()
```
2022-05-14 02:38:55 +08:00
2022-06-02 23:33:07 +08:00
`GROUP BY` converts the query to a summary, causing aggregation to happen:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-26 06:03:37 +08:00
flights |>
group_by(dest) |>
summarise(dep_delay = mean(dep_delay, na.rm = TRUE)) |>
show_query()
2022-05-11 06:26:04 +08:00
```
2022-05-26 06:03:37 +08:00
There are two important differences between dplyr verbs and SELECT clauses:
2022-05-31 21:40:22 +08:00
- In SQL, case doesn't matter: you can write `select`, `SELECT`, or even `SeLeCt`. In this book we'll stick with the common convention of writing SQL keywords in uppercase to distinguish them from table or variables names.
2022-06-02 23:33:07 +08:00
- In SQL, order matters: you must always write the clauses in the order `SELECT`, `FROM`, `WHERE`, `GROUP BY`, `ORDER BY`. Confusingly, this order doesn't match how the clauses actually evaluated which is first `FROM`, then `WHERE`, `GROUP BY`, `SELECT`, and `ORDER BY`.
2022-05-26 06:03:37 +08:00
2022-05-31 21:40:22 +08:00
The following sections explore each clause in more detail.
2022-05-26 06:03:37 +08:00
2022-05-26 23:55:12 +08:00
::: callout-note
2022-06-02 23:33:07 +08:00
Note that while SQL is a standard, it is extremely complex and no database follows it exactly.
While the main components that we'll focus on in this book are very similar between DBMSs, there are many minor variations.
Fortunately, dbplyr is designed to handle this problem and generates different translations for different databases.
2022-06-03 04:22:39 +08:00
It's not perfect, but it's continually improving, and if you hit a problem you can file an issue [on GitHub](https://github.com/tidyverse/dbplyr/issues/) to help us do better.
2022-05-26 23:55:12 +08:00
:::
2022-05-26 06:03:37 +08:00
### SELECT
2022-06-02 23:33:07 +08:00
The `SELECT` clause is the workhorse of queries and performs the same job as `select()`, `mutate()`, `rename()`, `relocate()`, and, as you'll learn in the next section, `summarize()`.
`select()`, `rename()`, and `relocate()` have very direct translations to `SELECT` as they just affect where a column appears (if at all) along with its name:
2022-05-14 02:38:55 +08:00
```{r}
2022-06-02 23:33:07 +08:00
planes |>
2022-08-04 21:00:38 +08:00
select(tailnum, type, manufacturer, model, year) |>
2022-05-26 06:03:37 +08:00
show_query()
2022-06-02 23:33:07 +08:00
planes |>
2022-08-04 21:00:38 +08:00
select(tailnum, type, manufacturer, model, year) |>
2022-06-02 23:33:07 +08:00
rename(year_built = year) |>
2022-05-26 06:03:37 +08:00
show_query()
2022-06-02 23:33:07 +08:00
planes |>
2022-08-04 21:00:38 +08:00
select(tailnum, type, manufacturer, model, year) |>
2022-06-02 23:33:07 +08:00
relocate(manufacturer, model, .before = type) |>
2022-05-26 06:03:37 +08:00
show_query()
2022-05-14 02:38:55 +08:00
```
2022-06-01 12:05:59 +08:00
This example also shows you how SQL does renaming.
In SQL terminology renaming is called **aliasing** and is done with `AS`.
Note that unlike with `mutate()`, the old name is on the left and the new name is on the right.
2022-05-14 02:38:55 +08:00
2022-08-04 21:00:38 +08:00
::: callout-note
In the examples above note that `"year"` and `"type"` are wrapped in double quotes.
That's because these are **reserved words** in duckdb, so dbplyr quotes them to avoid any potential confusion between column/table names and SQL operators.
When working with other databases you're likely to see every variable name quotes because only a handful of client packages, like duckdb, know what all the reserved words are, so they quote everything to be safe.
``` sql
SELECT "tailnum", "type", "manufacturer", "model", "year"
FROM "planes"
```
Some other database systems use backticks instead of quotes:
``` sql
SELECT `tailnum`, `type`, `manufacturer`, `model`, `year`
FROM `planes`
```
:::
2022-06-02 23:33:07 +08:00
The translations for `mutate()` are similarly straightforward: each variable becomes a new expression in `SELECT`:
2022-05-14 02:38:55 +08:00
```{r}
2022-08-04 21:00:38 +08:00
flights |>
2022-06-02 23:33:07 +08:00
mutate(
2022-08-04 21:00:38 +08:00
speed = distance / (air_time / 60)
2022-06-02 23:33:07 +08:00
) |>
2022-05-14 02:38:55 +08:00
show_query()
```
2022-05-31 21:40:22 +08:00
We'll come back to the translation of individual components (like `/`) in @sec-sql-expressions.
2022-08-04 21:00:38 +08:00
### FROM
2022-05-26 23:55:12 +08:00
2022-08-04 21:00:38 +08:00
The `FROM` clause defines the data source.
It's going to be rather uninteresting for a little while, because we're just using single tables.
You'll see more complex examples once we hit the join functions.
2022-05-26 23:55:12 +08:00
2022-05-11 06:26:04 +08:00
### GROUP BY
2022-08-04 21:00:38 +08:00
`group_by()` is translated to the `GROUP BY`[^databases-6] clause and `summarise()` is translated to the `SELECT` clause:
2022-05-31 21:40:22 +08:00
2022-08-04 21:00:38 +08:00
[^databases-6]: This is no coincidence: the dplyr function name was inspired by the SQL clause.
2022-05-11 06:26:04 +08:00
```{r}
diamonds_db |>
group_by(cut) |>
summarise(
n = n(),
2022-05-26 06:03:37 +08:00
avg_price = mean(price, na.rm = TRUE)
2022-05-11 06:26:04 +08:00
) |>
show_query()
```
2022-05-31 21:40:22 +08:00
We'll come back to what's happening with translation `n()` and `mean()` in @sec-sql-expressions.
2022-05-26 06:03:37 +08:00
### WHERE
2022-05-31 21:40:22 +08:00
`filter()` is translated to the `WHERE` clause:
2022-05-26 06:03:37 +08:00
```{r}
flights |>
filter(dest == "IAH" | dest == "HOU") |>
show_query()
flights |>
filter(arr_delay > 0 & arr_delay < 20) |>
show_query()
```
2022-05-31 21:40:22 +08:00
There are a few important details to note here:
2022-05-26 06:03:37 +08:00
2022-05-31 21:40:22 +08:00
- `|` becomes `OR` and `&` becomes `AND`.
- SQL uses `=` for comparison, not `==`. SQL doesn't have assignment, so there's no potential for confusion there.
2022-06-02 23:33:07 +08:00
- SQL uses only `''` for strings, not `""`. In SQL, `""` is used to identify variables, like R's ``` `` ```.
2022-05-31 21:40:22 +08:00
Another useful SQL operator is `IN`, which is very close to R's `%in%`:
2022-05-26 06:03:37 +08:00
```{r}
flights |>
filter(dest %in% c("IAH", "HOU")) |>
show_query()
```
2022-05-31 21:40:22 +08:00
SQL uses `NULL` instead of `NA`.
`NULL`s behave similarly to `NA`s.
The main difference is that while they're "infectious" in comparisons and arithmetic, they are silently dropped when summarizing.
2022-08-04 21:00:38 +08:00
dbplyr will remind you about this behavior the first time you hit it:
2022-05-31 21:40:22 +08:00
```{r}
flights |>
group_by(dest) |>
summarise(delay = mean(arr_delay))
```
2022-08-10 00:43:12 +08:00
If you want to learn more about how NULLs work, you might enjoy "[*Three valued logic*](https://modern-sql.com/concept/three-valued-logic)" by Markus Winand.
2022-06-01 20:53:33 +08:00
2022-06-02 23:33:07 +08:00
In general, you can work with `NULL`s using the functions you'd use for `NA`s in R:
2022-05-26 06:03:37 +08:00
```{r}
flights |>
filter(!is.na(dep_delay)) |>
show_query()
```
2022-05-31 21:40:22 +08:00
This SQL query illustrates one of the drawbacks of dbplyr: while the SQL is correct, it isn't as simple as you might write by hand.
In this case, you could drop the parentheses and use a special operator that's easier to read:
2022-05-11 06:26:04 +08:00
2022-05-26 06:03:37 +08:00
``` sql
WHERE "dep_delay" IS NOT NULL
```
2022-05-11 06:26:04 +08:00
2022-08-04 21:00:38 +08:00
Note that if you `filter()` a variable that you created using a summarize, dbplyr will generate a `HAVING` clause, rather than a `FROM` clause.
This is a one of the idiosyncracies of SQL created because `WHERE` is evaluated before `SELECT`, so it needs another clause that's evaluated afterwards.
```{r}
diamonds_db |>
group_by(cut) |>
summarise(n = n()) |>
filter(n > 100) |>
show_query()
```
2022-05-11 06:26:04 +08:00
### ORDER BY
2022-05-31 21:40:22 +08:00
Ordering rows involves a straightforward translation from `arrange()` to the `ORDER BY` clause:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-26 06:03:37 +08:00
flights |>
arrange(year, month, day, desc(dep_delay)) |>
2022-05-11 06:26:04 +08:00
show_query()
```
2022-06-02 23:33:07 +08:00
Notice how `desc()` is translated to `DESC`: this is one of the many dplyr functions whose name was directly inspired by SQL.
2022-05-14 02:38:55 +08:00
2022-05-11 06:26:04 +08:00
### Subqueries
2022-05-31 21:40:22 +08:00
Sometimes it's not possible to translate a dplyr pipeline into a single `SELECT` statement and you need to use a subquery.
A **subquery** is just a query used as a data source in the `FROM` clause, instead of the usual table.
2022-06-01 12:05:59 +08:00
dbplyr typically uses subqueries to work around limitations of SQL.
2022-05-31 21:40:22 +08:00
For example, expressions in the `SELECT` clause can't refer to columns that were just created.
2022-06-02 23:33:07 +08:00
That means that the following (silly) dplyr pipeline needs to happen in two steps: the first (inner) query computes `year1` and then the second (outer) query can compute `year2`.
2022-05-11 06:26:04 +08:00
```{r}
2022-05-31 21:40:22 +08:00
flights |>
2022-05-11 06:26:04 +08:00
mutate(
2022-05-31 21:40:22 +08:00
year1 = year + 1,
year2 = year1 + 1
2022-05-11 06:26:04 +08:00
) |>
show_query()
```
2022-05-31 21:40:22 +08:00
You'll also see this if you attempted to `filter()` a variable that you just created.
2022-06-02 23:33:07 +08:00
Remember, even though `WHERE` is written after `SELECT`, it's evaluated before it, so we need a subquery in this (silly) example:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-31 21:40:22 +08:00
flights |>
mutate(year1 = year + 1) |>
filter(year1 == 2014) |>
2022-05-11 06:26:04 +08:00
show_query()
```
2022-05-31 21:40:22 +08:00
Sometimes dbplyr will create a subquery where it's not needed because it doesn't yet know how to optimize that translation.
2022-06-02 23:33:07 +08:00
As dbplyr improves over time, these cases will get rarer but will probably never go away.
2022-05-11 06:26:04 +08:00
2022-05-24 23:12:41 +08:00
### Joins
2022-05-11 06:26:04 +08:00
2022-05-31 21:40:22 +08:00
If you're familiar with dplyr's joins, SQL joins are very similar.
Here's a simple example:
2022-05-11 06:26:04 +08:00
```{r}
2022-05-26 06:03:37 +08:00
flights |>
2022-06-02 23:33:07 +08:00
left_join(planes |> rename(year_built = year), by = "tailnum") |>
2022-05-26 06:03:37 +08:00
show_query()
2022-05-11 06:26:04 +08:00
```
2022-05-31 21:40:22 +08:00
The main thing to notice here is the syntax: SQL joins use sub-clauses of the `FROM` clause to bring in additional tables, using `ON` to define how the tables are related.
dplyr's names for these functions are so closely connected to SQL that you can easily guess the equivalent SQL for `inner_join()`, `right_join()`, and `full_join():`
2022-05-26 23:55:12 +08:00
``` sql
2022-05-31 21:40:22 +08:00
SELECT flights.*, "type", manufacturer, model, engines, seats, speed
FROM flights
INNER JOIN planes ON (flights.tailnum = planes.tailnum)
2022-05-26 23:55:12 +08:00
SELECT flights.*, "type", manufacturer, model, engines, seats, speed
FROM flights
RIGHT JOIN planes ON (flights.tailnum = planes.tailnum)
2022-05-26 06:03:37 +08:00
2022-05-26 23:55:12 +08:00
SELECT flights.*, "type", manufacturer, model, engines, seats, speed
FROM flights
FULL JOIN planes ON (flights.tailnum = planes.tailnum)
2022-05-11 06:26:04 +08:00
```
2022-06-02 23:33:07 +08:00
You're likely to need many joins when working with data from a database.
That's because database tables are often stored in a highly normalized form, where each "fact" is stored in a single place and to keep a complete dataset for analysis you need to navigate a complex network of tables connected by primary and foreign keys.
If you hit this scenario, the [dm package](https://cynkra.github.io/dm/), by Tobias Schieferdecker, Kirill Müller, and Darko Bergant, is a life saver.
It can automatically determine the connections between tables using the constraints that DBAs often supply, visualize the connections so you can see what's going on, and generate the joins you need to connect one table to another.
2022-05-11 06:26:04 +08:00
2022-05-26 23:55:12 +08:00
### Other verbs
2022-05-31 21:40:22 +08:00
dbplyr also translates other verbs like `distinct()`, `slice_*()`, and `intersect()`, and a growing selection of tidyr functions like `pivot_longer()` and `pivot_wider()`.
2022-05-26 23:55:12 +08:00
The easiest way to see the full set of what's currently available is to visit the dbplyr website: <https://dbplyr.tidyverse.org/reference/>.
2022-05-26 06:03:37 +08:00
2022-06-02 23:33:07 +08:00
### Exercises
1. What is `distinct()` translated to?
How about `head()`?
2. Explain what each of the following SQL queries do and try recreate them using dbplyr.
``` sql
SELECT *
FROM flights
WHERE dep_delay < arr_delay
SELECT *, distance / (airtime / 60) AS speed
FROM flights
```
2022-05-26 23:55:12 +08:00
## Function translations {#sec-sql-expressions}
2022-05-26 06:03:37 +08:00
2022-06-02 23:33:07 +08:00
So far we've focused on the big picture of how dplyr verbs are translated to the clauses of a query.
2022-06-01 20:53:33 +08:00
Now we're going to zoom in a little and talk about the translation of the R functions that work with individual columns, e.g. what happens when you use `mean(x)` in a `summarize()`?
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
To help see what's going on, we'll use a couple of little helper functions that run a `summarise()` or `mutate()` and show the generated SQL.
That will make it a easier to explore a few variations and see how summaries and transformations can differ.
2022-05-24 23:12:41 +08:00
2022-05-26 23:55:12 +08:00
```{r}
2022-06-01 20:53:33 +08:00
summarize_query <- function(df, ...) {
2022-05-26 23:55:12 +08:00
df |>
summarise(...) |>
show_query()
}
2022-06-01 20:53:33 +08:00
mutate_query <- function(df, ...) {
2022-05-26 23:55:12 +08:00
df |>
2022-06-01 20:53:33 +08:00
mutate(..., .keep = "none") |>
2022-05-26 23:55:12 +08:00
show_query()
}
```
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
Let's dive in with some summaries!
Looking at the code below you'll notice that some summary functions, like `mean()`, have a relatively simple translation while others, like `median()`, are much more complex.
2022-06-02 23:33:07 +08:00
The complexity is typically higher for operations that are common in statistics but less common in databases.
2022-06-01 20:53:33 +08:00
2022-05-11 06:26:04 +08:00
```{r}
2022-06-01 20:53:33 +08:00
flights |>
group_by(year, month, day) |>
summarize_query(
mean = mean(arr_delay, na.rm = TRUE),
median = median(arr_delay, na.rm = TRUE)
)
2022-05-11 06:26:04 +08:00
```
2022-06-02 23:33:07 +08:00
The translation of summary functions becomes more complicated when you use them inside a `mutate()` because they have to turn into a window function.
In SQL, you turn an ordinary aggregation function into a window function by adding `OVER` after it:
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
```{r}
flights |>
group_by(year, month, day) |>
mutate_query(
mean = mean(arr_delay, na.rm = TRUE),
)
```
2022-05-11 06:26:04 +08:00
In SQL, the `GROUP BY` clause is used exclusively for summary so here you can see that the grouping has moved to the `PARTITION BY` argument to `OVER`.
2022-05-11 06:26:04 +08:00
Window functions include all functions that look forward or backwards, like `lead()` and `lag()`:
2022-06-01 20:53:33 +08:00
```{r}
flights |>
group_by(dest) |>
arrange(time_hour) |>
mutate_query(
lead = lead(arr_delay),
lag = lag(arr_delay)
)
```
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
Here it's important to `arrange()` the data, because SQL tables have no intrinsic order.
In fact, if you don't use `arrange()` you might get the rows back in a different order every time!
2022-06-02 23:33:07 +08:00
Notice for window functions, the ordering information is repeated: the `ORDER BY` clause of the main query doesn't automatically apply to window functions.
2022-05-11 06:26:04 +08:00
2022-06-02 23:33:07 +08:00
Another important SQL function is `CASE WHEN`. It's used as the translation of `if_else()` and `case_when()`, the dplyr function that it directly inspired.
2022-06-01 20:53:33 +08:00
Here's a couple of simple examples:
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
```{r}
flights |>
mutate_query(
2022-08-04 21:00:38 +08:00
description = if_else(arr_delay > 0, "delayed", "on-time")
2022-06-01 20:53:33 +08:00
)
flights |>
mutate_query(
description =
case_when(
arr_delay < -5 ~ "early",
arr_delay < 5 ~ "on-time",
arr_delay >= 5 ~ "late"
)
)
```
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
`CASE WHEN` is also used for some other functions that don't have a direct translation from R to SQL.
A good example of this is `cut()`:
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
```{r}
flights |>
mutate_query(
description = cut(
arr_delay,
breaks = c(-Inf, -5, 5, Inf),
labels = c("early", "on-time", "late")
)
)
```
2022-05-11 06:26:04 +08:00
2022-06-01 20:53:33 +08:00
dbplyr also translates common string and date-time manipulation functions, which you can learn about in `vignette("translation-function", package = "dbplyr")`.
dbplyr's translations are certainly not perfect, and there are many R functions that aren't translated yet, but dbplyr does a surprisingly good job covering the functions that you'll use most of the time.
2022-06-02 23:33:07 +08:00
### Learning more
If you've finished this chapter and would like to learn more about SQL.
2022-08-10 00:43:12 +08:00
We have two recommendations:
2022-06-02 23:33:07 +08:00
- [*SQL for Data Scientists*](https://sqlfordatascientists.com) by Renée M. P. Teate is an introduction to SQL designed specifically for the needs of data scientists, and includes examples of the sort of highly interconnected data you're likely to encounter in real organisations.
- [*Practical SQL*](https://www.practicalsql.com) by Anthony DeBarros is written from the perspective of a data journalist (a data scientist specialized in telling compelling stories) and goes into more detail about getting your data into a database and running your own DBMS.