r4ds/logicals.Rmd

419 lines
14 KiB
Plaintext
Raw Normal View History

# Logical vectors {#logicals}
2021-03-04 01:13:14 +08:00
2021-05-04 21:10:39 +08:00
```{r, results = "asis", echo = FALSE}
status("drafting")
```
2021-03-04 01:13:14 +08:00
## Introduction
2021-04-19 20:56:29 +08:00
2022-03-17 22:46:35 +08:00
In this chapter, you'll learn useful tools for working with logical vectors.
2022-03-18 03:15:24 +08:00
Logical vectors are the simplest type of vector because each element can only be one of three possible values: `TRUE`, `FALSE`, and `NA`.
You'll find logical vectors directly in data relatively rarely, but despite that they're extremely powerful because you'll frequently create them during data analysis.
We'll begin with the most common way of creating logical vectors: numeric comparisons.
Then we'll talk about using Boolean algebra to combine different logical vectors, and some useful summaries for logical vectors.
We'll finish off with some other tool for making conditional changes
2021-04-19 20:56:29 +08:00
2022-02-05 02:27:20 +08:00
### Prerequisites
In this chapter, we'll continue to draw inspiration from the nyclights13 dataset.
```{r setup, message = FALSE}
2021-04-19 22:31:38 +08:00
library(tidyverse)
library(nycflights13)
```
But as we start to discuss more tools, there won't always be a perfect example.
So from this chapter on we'll start to use more abstract examples where we create a vector with `c()`, and then manipulate it in various ways.
This will make it easier to explain the general point without having to construct a full example.
It does make it a little harder to apply directly to your data problems, but remember that you can do these same manipulations with a vector inside a data frame using `mutate()` and friends.
2022-02-05 02:27:20 +08:00
```{r}
x <- c(1, 2, 3, 5, 7, 11, 13)
x * 2
2022-03-17 22:46:35 +08:00
df <- tibble(
x = c(1, 2, 3, 5, 7, 11, 13)
)
df |>
mutate(y = x * 2)
```
## Comparisons
2022-03-17 22:46:35 +08:00
A very common way to create a logical vector is via a numeric comparison with `<`, `<=`, `>`, `>=`, `!=`, and `==`.
You'll learn other ways to create them in later chapters dealing with strings and dates.
So far, we've mostly created logical variables implicitly within `filter()` --- they are computed, used, and then throw away.
For example, the following filter finds all day time departures that leave roughly on time:
2022-03-18 03:15:24 +08:00
```{r}
flights |>
filter(dep_time > 600 & dep_time < 2000 & abs(arr_delay) < 20)
```
But it's useful to know that this is a shortcut and you can explicitly create the underlying logical variables with `mutate()`:
2022-03-18 03:15:24 +08:00
```{r}
flights |>
mutate(
daytime = dep_time > 600 & dep_time < 2000,
approx_ontime = abs(arr_delay) < 20,
.keep = "used"
)
```
This is useful because it allows you to name components, which can made the code easier to read, and it allows you to double-check the intermediate steps.
This is a particularly useful technique when you're doing more complicated Boolean algebra, as you'll learn about in the next section.
2022-03-18 03:15:24 +08:00
So the initial filter could also be written as:
```{r, results = FALSE}
2022-03-18 03:15:24 +08:00
flights |>
mutate(
daytime = dep_time > 600 & dep_time < 2000,
approx_ontime = abs(arr_delay) < 20,
) |>
filter(daytime & approx_ontime)
```
### Floating point comparison
2022-03-17 22:46:35 +08:00
Beware when using `==` with numbers as results might surprise you!
It looks like this vector contains the numbers 1 and 2:
2022-03-18 03:15:24 +08:00
```{r}
x <- c(1 / 49 * 49, sqrt(2) ^ 2)
x
2022-03-18 03:15:24 +08:00
```
But if you test them for equality, you surprisingly get `FALSE`:
2022-03-17 22:46:35 +08:00
```{r}
x == c(1, 2)
2022-03-17 22:46:35 +08:00
```
2022-03-18 03:15:24 +08:00
That's because computers use finite precision arithmetic (they obviously can't store an infinite number of digits!) so in most cases, the number number you see is an actually approximation.
R usually rounds these numbers to avoid displaying a bunch of usually unimportant digits.
To see the details you can call `print()` with the the `digits`[^logicals-1] argument.
R normally calls print automatically for you (i.e. `x` is a shortcut for `print(x)`), but calling it explicitly is useful if you want to provide other arguments:
[^logicals-1]: A floating point number can hold roughly 16 decimal digits; the precise number is surprisingly complicated and depends on the number.
2022-03-17 22:46:35 +08:00
```{r}
print(x, digits = 16)
2022-03-17 22:46:35 +08:00
```
Now that you've seen why `==` is failing, what can you do about it?
One option is to use `round()` to round to any number of digits, or instead of `==`, use `dplyr::near()`, which does the comparison with a small amount of tolerance:
2022-03-17 22:46:35 +08:00
```{r}
near(x, c(1, 2))
2022-03-17 22:46:35 +08:00
```
2021-04-19 20:56:29 +08:00
### Missing values {#na-comparison}
Missing values represent the unknown so they missing values are "contagious": almost any operation involving an unknown value will also be unknown:
```{r}
NA > 5
10 == NA
```
The most confusing result is this one:
```{r}
NA == NA
```
It's easiest to understand why this is true with a bit more context:
```{r}
# Let x be Mary's age. We don't know how old she is.
x <- NA
# Let y be John's age. We don't know how old he is.
y <- NA
# Are John and Mary the same age?
x == y
# We don't know!
```
So if you want to find all flights with `dep_time` is missing, the following code won't work because `dep_time == NA` will yield a `NA` for every single row, and `filter()` automatically drops missing values:
```{r}
flights |>
filter(dep_time == NA)
```
Instead we'll need a new too: `is.na()`.
2022-03-18 03:15:24 +08:00
### `is.na()`
There's one other very useful way to create logical vectors: `is.na()`.
This takes any type of vector and returns `TRUE` is the value is `NA`, and `FALSE` otherwise:
```{r}
is.na(c(TRUE, NA, FALSE))
is.na(c(1, NA, 3))
is.na(c("a", NA, "b"))
```
We can use `is.na()` to find all the rows with a missing `dep_time`:
```{r}
flights |>
filter(is.na(dep_time))
```
It can also be useful in `arrange()`, because by default, `arrange()` puts all the missing values at the end.
You can override this default by first sorting by `is.na()`:
2022-03-18 03:15:24 +08:00
```{r}
flights |>
arrange(arr_delay)
flights |>
arrange(desc(is.na(arr_delay)), arr_delay)
2022-03-18 03:15:24 +08:00
```
2022-03-17 22:46:35 +08:00
### Exercises
1. How does `dplyr::near()` work? Read the source code to find out.
2. Use `mutate()`, `is.na()`, and `count()` together to describe how the missing values in `dep_time`, `sched_dep_time` and `dep_delay` are connected.
2022-03-17 22:46:35 +08:00
## Boolean algebra
2022-02-05 02:27:20 +08:00
Once you have multiple logical vectors, you can combine them together using Boolean algebra.
In R, `&` is "and", `|` is "or", and `!` is "not", and `xor()` is exclusive or[^logicals-2].
2022-03-18 03:15:24 +08:00
Figure \@ref(fig:bool-ops) shows the complete set of Boolean operations and how they work.
2021-04-19 20:56:29 +08:00
[^logicals-2]: That is, `xor(x, y)` is true if x is true, or y is true, but not both.
This is how we usually use "or" In English.
Both is not usually an acceptable answer to the question "would you like ice cream or cake?".
2022-02-05 02:27:20 +08:00
```{r bool-ops}
#| echo: false
2022-03-18 03:15:24 +08:00
#| out.width: NULL
2022-02-05 02:27:20 +08:00
#| fig.cap: >
#| Complete set of boolean operations. `x` is the left-hand
#| circle, `y` is the right-hand circle, and the shaded region show
#| which parts each operator selects."
#| fig.alt: >
#| Six Venn diagrams, each explaining a given logical operator. The
#| circles (sets) in each of the Venn diagrams represent x and y. 1. y &
#| !x is y but none of x, x & y is the intersection of x and y, x & !y is
#| x but none of y, x is all of x none of y, xor(x, y) is everything
#| except the intersection of x and y, y is all of y none of x, and
#| x | y is everything.
knitr::include_graphics("diagrams/transform.png", dpi = 270)
2021-04-19 20:56:29 +08:00
```
2022-03-18 03:15:24 +08:00
As well as `&` and `|`, R also has `&&` and `||`.
Don't use them in dplyr functions!
These are called short-circuiting operators and only ever return a single `TRUE` or `FALSE`.
They're important for programming so you'll learn more about them in Section \@ref(conditional-execution).
2021-04-19 20:56:29 +08:00
The following code finds all flights that departed in November or December:
```{r, eval = FALSE}
2022-03-18 03:15:24 +08:00
flights |>
filter(month == 11 | month == 12)
2021-04-19 20:56:29 +08:00
```
2022-02-05 02:27:20 +08:00
Note that the order of operations doesn't work like English.
2022-03-18 03:15:24 +08:00
You can't think "find all flights that departed in November or December" and write `flights |> filter(month == 11 | 12)`.
This code will not error, but it will do something rather confusing.
First R evaluates `11 | 12` which is equivalent to `TRUE | TRUE`, which returns `TRUE`.
2022-02-05 02:27:20 +08:00
Then it evaluates `month == TRUE`.
2022-03-18 03:15:24 +08:00
Since month is numeric, this is equivalent to `month == 1`, so `flights |> filter(month == 11 | 12)` returns all flights in January!
### `%in%`
2021-04-19 20:56:29 +08:00
2022-03-18 03:15:24 +08:00
An easy way to avoid this issue is to use `%in%`.
2022-02-05 02:27:20 +08:00
`x %in% y` returns a logical vector the same length as `x` that is `TRUE` whenever a value in `x` is anywhere in `y` .
```{r}
letters[1:10] %in% c("a", "e", "i", "o", "u")
```
2022-03-18 03:15:24 +08:00
So we could instead write:
2021-04-19 20:56:29 +08:00
```{r, eval = FALSE}
2022-03-18 03:15:24 +08:00
flights |>
filter(month %in% c(11, 12))
2021-04-19 20:56:29 +08:00
```
Note the `%in%` obeys different rules for `NA` to `==`.
2021-04-19 20:56:29 +08:00
```{r}
2022-03-18 03:15:24 +08:00
flights |>
filter(dep_time %in% c(NA, 0800))
2021-04-19 20:56:29 +08:00
```
### Missing values {#na-boolean}
2022-03-18 03:15:24 +08:00
The rules for missing values in Boolean algebra are a little tricky to explain because they seem inconsistent at first glance:
```{r}
NA & c(TRUE, FALSE, NA)
NA | c(TRUE, FALSE, NA)
```
2021-04-19 20:56:29 +08:00
To understand what's going on, think about `NA | TRUE`.
If a logical is `NA`, than means it could either be `TRUE` or `FALSE`.
`TRUE | TRUE` and `FALSE | TRUE` are both `TRUE`, so `NA | TRUE` must also be `TRUE`.
Similar reasoning applies with `NA & FALSE`.
### Exercises
2021-04-19 20:56:29 +08:00
1. Find all flights where `arr_delay` is missing but `dep_delay` is not. Find all flights where neither `arr_time` nor `sched_arr_time` are missing, but `arr_delay` is.
2022-03-18 03:15:24 +08:00
## Summaries
There are four particularly useful summary functions for logical vectors: they all take a vector of logical values and return a single value, making them a good fit for use in `summarise()`.
`any()` and `all()` --- `any()` will return if there's at least one `TRUE`, `all()` will return `TRUE` if all values are `TRUE`.
Like all summary functions, they'll return `NA` if there are any missing values present, and like usual you can make the missing values go away with `na.rm = TRUE`.
We could use this to see if there were any days where every flight was delayed:
2021-04-19 20:56:29 +08:00
2022-02-05 02:27:20 +08:00
```{r}
2022-03-18 03:15:24 +08:00
not_cancelled <- flights |> filter(!is.na(dep_delay), !is.na(arr_delay))
not_cancelled |>
group_by(year, month, day) |>
filter(all(arr_delay >= 0))
2022-02-05 02:27:20 +08:00
```
2021-04-19 20:56:29 +08:00
2022-03-18 03:15:24 +08:00
`sum()` and `mean()` are particularly useful with logical vectors because when you use a logical vector in a numeric context, `TRUE` becomes 1 and `FALSE` becomes 0.
That means that `sum(x)` gives the number of `TRUE`s in `x` and `mean(x)` gives the proportion of `TRUE`s.
That lets us find the day with the highest proportion of delayed flights:
```{r}
not_cancelled |>
group_by(year, month, day) |>
summarise(prop_delayed = mean(arr_delay > 0)) |>
arrange(desc(prop_delayed))
```
2021-04-19 22:31:38 +08:00
2022-03-18 03:15:24 +08:00
Or we could ask how many flights left before 5am, which usually are flights that were delayed from the previous day:
2021-04-19 20:56:29 +08:00
2022-02-05 02:27:20 +08:00
```{r}
2022-03-18 03:15:24 +08:00
not_cancelled |>
group_by(year, month, day) |>
summarise(n_early = sum(dep_time < 500)) |>
arrange(desc(n_early))
2022-02-05 02:27:20 +08:00
```
You can also use logical vectors inside summaries:
2022-03-17 22:46:35 +08:00
2022-03-18 03:15:24 +08:00
```{r}
not_cancelled |>
group_by(year, month, day) |>
summarise(
ahead = mean(arr_delay[arr_delay > 0]),
behind = mean(arr_delay[arr_delay < 0]),
)
2022-03-18 03:15:24 +08:00
```
### Exercises
2022-03-18 03:15:24 +08:00
1. For each plane, count the number of flights before the first delay of greater than 1 hour.
2. What does `prod()` return when applied to a logical vector? What logical summary function is it equivalent to? What does `min()` return applied to a logical vector? What logical summary function is it equivalent to?
2022-03-17 22:46:35 +08:00
## Transformations
2022-03-17 22:46:35 +08:00
2022-03-18 03:15:24 +08:00
### Conditional outputs
2022-02-05 02:27:20 +08:00
If you want to use one value when a condition is true and another value when it's `FALSE`, you can use `if_else()`[^logicals-3].
2022-02-05 02:27:20 +08:00
[^logicals-3]: This is equivalent to the base R function `ifelse`.
2022-02-05 02:27:20 +08:00
There are two main advantages of `if_else()`over `ifelse()`: you can choose what should happen to missing values, and `if_else()` is much more likely to give you a meaningful error message if you use the wrong type of variable.
```{r}
df <- data.frame(
date = as.Date("2020-01-01") + 0:6,
balance = c(100, 50, 25, -25, -50, 30, 120)
)
2022-02-24 03:15:52 +08:00
df |> mutate(status = if_else(balance < 0, "overdraft", "ok"))
2022-02-05 02:27:20 +08:00
```
If you start to nest multiple sets of `if_else`s, I'd suggest switching to `case_when()` instead.
`case_when()` has a special syntax: it takes pairs that look like `condition ~ output`.
`condition` must evaluate to a logical vector; when it's `TRUE`, output will be used.
2021-04-19 20:56:29 +08:00
2022-02-05 02:27:20 +08:00
```{r}
2022-02-24 03:15:52 +08:00
df |>
2022-02-05 02:27:20 +08:00
mutate(
status = case_when(
balance == 0 ~ "no money",
balance < 0 ~ "overdraft",
balance > 0 ~ "ok"
)
)
```
(Note that I usually add spaces to make the outputs line up so it's easier to scan)
If none of the cases match, the output will be missing:
```{r}
x <- 1:10
case_when(
x %% 2 == 0 ~ "even",
)
```
You can create a catch all value by using `TRUE` as the condition:
```{r}
case_when(
x %% 2 == 0 ~ "even",
TRUE ~ "odd"
)
```
If multiple conditions are `TRUE`, the first is used:
```{r}
case_when(
x < 5 ~ "< 5",
x < 3 ~ "< 3",
)
```
### Cumulative functions
Another useful pair of functions are cumulative any, `cumany()`, and cumulative all, `cumall()`.
`cumany()` will be `TRUE` after it encounters the first `TRUE`, and `cumall()` will be `FALSE` after it encounters its first `FALSE`.
```{r}
cumany(c(FALSE, FALSE, TRUE, TRUE, FALSE, TRUE))
cumall(c(TRUE, FALSE, TRUE, TRUE, FALSE, TRUE))
```
These are particularly useful in conjunction with `filter()` because they allow you to select rows:
- Before the first `FALSE` with `cumall(x)`.
- Before the first `TRUE` with `cumall(!x)`.
- After the first `TRUE` with `cumany(x)`.
- After the first `FALSE` with `cumany(!x)`.
If you imagine some data about a bank balance, then these functions allow you t
```{r}
df <- data.frame(
date = as.Date("2020-01-01") + 0:6,
balance = c(100, 50, 25, -25, -50, 30, 120)
)
# all rows after first overdraft
df |> filter(cumany(balance < 0))
# all rows until first overdraft
df |> filter(cumall(!(balance < 0)))
```
###
2022-02-05 02:27:20 +08:00
2022-03-17 22:46:35 +08:00
##