1093 lines
75 KiB
HTML
1093 lines
75 KiB
HTML
|
<section data-type="chapter" id="chp-iteration">
|
|||
|
<h1><span id="sec-iteration" class="quarto-section-identifier d-none d-lg-block"><span class="chapter-title">Iteration</span></span></h1><div data-type="note"><div class="callout-body d-flex">
|
|||
|
<div class="callout-icon-container">
|
|||
|
<i class="callout-icon"/>
|
|||
|
</div>
|
|||
|
|
|||
|
</div>
|
|||
|
|
|||
|
<p>You are reading the work-in-progress second edition of R for Data Science. This chapter should be readable but is currently undergoing final polishing. You can find the complete first edition at <a href="https://r4ds.had.co.nz" class="uri">https://r4ds.had.co.nz</a>.</p></div>
|
|||
|
|
|||
|
<section id="introduction" data-type="sect1">
|
|||
|
<h1>
|
|||
|
Introduction</h1>
|
|||
|
<p>In this chapter, you’ll learn tools for iteration, repeatedly performing the same action on different objects. Iteration in R generally tends to look rather different from other programming languages because so much of it is implicit and we get it for free. For example, if you want to double a numeric vector <code>x</code> in R, you can just write <code>2 * x</code>. In most other languages, you’d need to explicitly double each element of x using some sort of for loop.</p>
|
|||
|
<p>This book has already given you a small but powerful number of tools that perform the same action for multiple “things”:</p>
|
|||
|
<ul><li>
|
|||
|
<code><a href="#chp-https://ggplot2.tidyverse.org/reference/facet_wrap" data-type="xref">#chp-https://ggplot2.tidyverse.org/reference/facet_wrap</a></code> and <code><a href="#chp-https://ggplot2.tidyverse.org/reference/facet_grid" data-type="xref">#chp-https://ggplot2.tidyverse.org/reference/facet_grid</a></code> draws a plot for each subset.</li>
|
|||
|
<li>
|
|||
|
<code><a href="#chp-https://dplyr.tidyverse.org/reference/group_by" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/group_by</a></code> plus <code><a href="#chp-https://dplyr.tidyverse.org/reference/summarise" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/summarise</a></code> computes a summary statistics for each subset.</li>
|
|||
|
<li>
|
|||
|
<code><a href="#chp-https://tidyr.tidyverse.org/reference/unnest_wider" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/unnest_wider</a></code> and <code><a href="#chp-https://tidyr.tidyverse.org/reference/unnest_longer" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/unnest_longer</a></code> create new rows and columns for each element of a list-column.</li>
|
|||
|
</ul><p>Now it’s time to learn some more general tools, often called <strong>functional programming</strong> tools because they are built around functions that take other functions as inputs. Learning functional programming can easily veer into the abstract, but in this chapter we’ll keep things concrete by focusing on three common tasks: modifying multiple columns, reading multiple files, and saving multiple objects.</p>
|
|||
|
|
|||
|
<section id="prerequisites" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Prerequisites</h2>
|
|||
|
<div data-type="important"><div class="callout-body d-flex">
|
|||
|
<div class="callout-icon-container">
|
|||
|
<i class="callout-icon"/>
|
|||
|
</div>
|
|||
|
|
|||
|
</div>
|
|||
|
|
|||
|
<p>This chapter relies on features only found in purrr 1.0.0 and dplyr 1.1.0, which are still in development. If you want to live life on the edge you can get the dev version with <code>devtools::install_github(c("tidyverse/purrr", "tidyverse/dplyr"))</code>.</p></div>
|
|||
|
|
|||
|
<p>In this chapter, we’ll focus on tools provided by dplyr and purrr, both core members of the tidyverse. You’ve seen dplyr before, but <a href="#chp-http://purrr.tidyverse.org/" data-type="xref">#chp-http://purrr.tidyverse.org/</a> is new. We’re going to use just a couple of purrr functions from in this chapter, but it’s a great package to explore as you improve your programming skills.</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">library(tidyverse)</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="sec-across" data-type="sect1">
|
|||
|
<h1>
|
|||
|
Modifying multiple columns</h1>
|
|||
|
<p>Imagine you have this simple tibble and you want to count the number of observations and compute the median of every column.</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df <- tibble(
|
|||
|
a = rnorm(10),
|
|||
|
b = rnorm(10),
|
|||
|
c = rnorm(10),
|
|||
|
d = rnorm(10)
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>You could do it with copy-and-paste:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df |> summarise(
|
|||
|
n = n(),
|
|||
|
a = median(a),
|
|||
|
b = median(b),
|
|||
|
c = median(c),
|
|||
|
d = median(d),
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 5
|
|||
|
#> n a b c d
|
|||
|
#> <int> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 10 -0.246 -0.287 -0.0567 0.144</pre>
|
|||
|
</div>
|
|||
|
<p>That breaks our rule of thumb to never copy and paste more than twice, and you can imagine that this will get very tedious if you have tens or even hundreds of columns. Instead you can use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df |> summarise(
|
|||
|
n = n(),
|
|||
|
across(a:d, median),
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 5
|
|||
|
#> n a b c d
|
|||
|
#> <int> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 10 -0.246 -0.287 -0.0567 0.144</pre>
|
|||
|
</div>
|
|||
|
<p><code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> has three particularly important arguments, which we’ll discuss in detail in the following sections. You’ll use the first two every time you use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>: the first argument, <code>.cols</code>, specifies which columns you want to iterate over, and the second argument, <code>.fns</code>, specifies what to do with each column. You can use the <code>.names</code> argument when you need additional control over the names of output columns, which is particularly important when you use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> with <code><a href="#chp-https://dplyr.tidyverse.org/reference/mutate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/mutate</a></code>. We’ll also discuss two important variations, <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> and <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>, which work with <code><a href="#chp-https://dplyr.tidyverse.org/reference/filter" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/filter</a></code>.</p>
|
|||
|
|
|||
|
<section id="selecting-columns-with-.cols" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Selecting columns with<code>.cols</code>
|
|||
|
</h2>
|
|||
|
<p>The first argument to <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>, <code>.cols</code>, selects the columns to transform. This uses the same specifications as <code><a href="#chp-https://dplyr.tidyverse.org/reference/select" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/select</a></code>, <a href="#sec-select" data-type="xref">#sec-select</a>, so you can use functions like <code><a href="#chp-https://tidyselect.r-lib.org/reference/starts_with" data-type="xref">#chp-https://tidyselect.r-lib.org/reference/starts_with</a></code> and <code><a href="#chp-https://tidyselect.r-lib.org/reference/starts_with" data-type="xref">#chp-https://tidyselect.r-lib.org/reference/starts_with</a></code> to select columns based on their name.</p>
|
|||
|
<p>There are two additional selection techniques that are particularly useful for <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>: <code><a href="#chp-https://tidyselect.r-lib.org/reference/everything" data-type="xref">#chp-https://tidyselect.r-lib.org/reference/everything</a></code> and <code>where()</code>. <code><a href="#chp-https://tidyselect.r-lib.org/reference/everything" data-type="xref">#chp-https://tidyselect.r-lib.org/reference/everything</a></code> is straightforward: it selects every (non-grouping) column:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df <- tibble(
|
|||
|
grp = sample(2, 10, replace = TRUE),
|
|||
|
a = rnorm(10),
|
|||
|
b = rnorm(10),
|
|||
|
c = rnorm(10),
|
|||
|
d = rnorm(10)
|
|||
|
)
|
|||
|
|
|||
|
df |>
|
|||
|
group_by(grp) |>
|
|||
|
summarise(across(everything(), median))
|
|||
|
#> # A tibble: 2 × 5
|
|||
|
#> grp a b c d
|
|||
|
#> <int> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 1 -0.0935 -0.0163 0.363 0.364
|
|||
|
#> 2 2 0.312 -0.0576 0.208 0.565</pre>
|
|||
|
</div>
|
|||
|
<p>Note grouping columns (<code>grp</code> here) are not included in <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>, because they’re automatically preserved by <code><a href="#chp-https://dplyr.tidyverse.org/reference/summarise" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/summarise</a></code>.</p>
|
|||
|
<p><code>where()</code> allows you to select columns based on their type:</p>
|
|||
|
<ul><li>
|
|||
|
<code>where(is.numeric)</code> selects all numeric columns.</li>
|
|||
|
<li>
|
|||
|
<code>where(is.character)</code> selects all string columns.</li>
|
|||
|
<li>
|
|||
|
<code>where(is.Date)</code> selects all date columns.</li>
|
|||
|
<li>
|
|||
|
<code>where(is.POSIXct)</code> selects all date-time columns.</li>
|
|||
|
<li>
|
|||
|
<code>where(is.logical)</code> selects all logical columns.</li>
|
|||
|
</ul><div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_types <- tibble(
|
|||
|
x1 = 1:3,
|
|||
|
x2 = runif(3),
|
|||
|
y1 = sample(letters, 3),
|
|||
|
y2 = c("banana", "apple", "egg")
|
|||
|
)
|
|||
|
|
|||
|
df_types |>
|
|||
|
summarise(across(where(is.numeric), mean))
|
|||
|
#> # A tibble: 1 × 2
|
|||
|
#> x1 x2
|
|||
|
#> <dbl> <dbl>
|
|||
|
#> 1 2 0.370
|
|||
|
|
|||
|
df_types |>
|
|||
|
summarise(across(where(is.character), str_flatten))
|
|||
|
#> # A tibble: 1 × 2
|
|||
|
#> y1 y2
|
|||
|
#> <chr> <chr>
|
|||
|
#> 1 kjh bananaappleegg</pre>
|
|||
|
</div>
|
|||
|
<p>Just like other selectors, you can combine these with Boolean algebra. For example, <code>!where(is.numeric)</code> selects all non-numeric columns and <code>starts_with("a") & where(is.logical)</code> selects all logical columns whose name starts with “a”.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="calling-a-single-function" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Calling a single function</h2>
|
|||
|
<p>The second argument to <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> defines how each column will be transformed. In simple cases, as above, this will be a single existing function. This is a pretty special feature of R: we’re passing one function (<code>median</code>, <code>mean</code>, <code>str_flatten</code>, …) to another function (<code>across</code>). This is one of the features that makes R a function programming language.</p>
|
|||
|
<p>It’s important to note that we’re passing this function to <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>, so <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> can call it, not calling it ourselves. That means the function name should never be followed by <code>()</code>. If you forget, you’ll get an error:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df |>
|
|||
|
group_by(grp) |>
|
|||
|
summarise(across(everything(), median()))
|
|||
|
#> Error in vapply(.x, .f, .mold, ..., USE.NAMES = FALSE): values must be length 1,
|
|||
|
#> but FUN(X[[1]]) result is length 0</pre>
|
|||
|
</div>
|
|||
|
<p>This error arises because you’re calling the function with no input, e.g.:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">median()
|
|||
|
#> Error in is.factor(x): argument "x" is missing, with no default</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="calling-multiple-functions" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Calling multiple functions</h2>
|
|||
|
<p>In more complex cases, you might want to supply additional arguments or perform multiple transformations. Lets motivate this problem with a simple example: what happens if we have some missing values in our data? <code><a href="#chp-https://rdrr.io/r/stats/median" data-type="xref">#chp-https://rdrr.io/r/stats/median</a></code> propagates those missing values, giving us a suboptimal output:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">rnorm_na <- function(n, n_na, mean = 0, sd = 1) {
|
|||
|
sample(c(rnorm(n - n_na, mean = mean, sd = 1), rep(NA, n_na)))
|
|||
|
}
|
|||
|
|
|||
|
df_miss <- tibble(
|
|||
|
a = rnorm_na(5, 1),
|
|||
|
b = rnorm_na(5, 1),
|
|||
|
c = rnorm_na(5, 2),
|
|||
|
d = rnorm(5)
|
|||
|
)
|
|||
|
df_miss |>
|
|||
|
summarise(
|
|||
|
across(a:d, median),
|
|||
|
n = n()
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 5
|
|||
|
#> a b c d n
|
|||
|
#> <dbl> <dbl> <dbl> <dbl> <int>
|
|||
|
#> 1 NA NA NA 0.704 5</pre>
|
|||
|
</div>
|
|||
|
<p>It would be nice if we could pass along <code>na.rm = TRUE</code> to <code><a href="#chp-https://rdrr.io/r/stats/median" data-type="xref">#chp-https://rdrr.io/r/stats/median</a></code> to remove these missing values. To do so, instead of calling <code><a href="#chp-https://rdrr.io/r/stats/median" data-type="xref">#chp-https://rdrr.io/r/stats/median</a></code> directly, we need to create a new function that calls <code><a href="#chp-https://rdrr.io/r/stats/median" data-type="xref">#chp-https://rdrr.io/r/stats/median</a></code> with the desired arguments:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
summarise(
|
|||
|
across(a:d, function(x) median(x, na.rm = TRUE)),
|
|||
|
n = n()
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 5
|
|||
|
#> a b c d n
|
|||
|
#> <dbl> <dbl> <dbl> <dbl> <int>
|
|||
|
#> 1 0.429 -0.721 -0.796 0.704 5</pre>
|
|||
|
</div>
|
|||
|
<p>This is a little verbose, so R comes with a handy shortcut: for this sort of throw away, or <strong>anonymous</strong><span data-type="footnote">Anonymous, because we never explicitly gave it a name with <code><-</code>. Another term programmers use for this is “lambda function”.</span>, function you can replace <code>function</code> with <code>\</code><span data-type="footnote">In older code you might see syntax that looks like <code>~ .x + 1</code>. This is another way to write anonymous functions but it only works inside tidyverse functions and always uses the variable name <code>.x</code>. We now recommend the base syntax, <code>\(x) x + 1</code>.</span>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
summarise(
|
|||
|
across(a:d, \(x) median(x, na.rm = TRUE)),
|
|||
|
n = n()
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>In either case, <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> effectively expands to the following code:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
summarise(
|
|||
|
a = median(a, na.rm = TRUE),
|
|||
|
b = median(b, na.rm = TRUE),
|
|||
|
c = median(c, na.rm = TRUE),
|
|||
|
d = median(d, na.rm = TRUE),
|
|||
|
n = n()
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>When we remove the missing values from the <code><a href="#chp-https://rdrr.io/r/stats/median" data-type="xref">#chp-https://rdrr.io/r/stats/median</a></code>, it would be nice to know just how many values we were removing. We can find that out by supplying two functions to <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>: one to compute the median and the other to count the missing values. You supply multiple functions by using a named list to <code>.fns</code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
summarise(
|
|||
|
across(a:d, list(
|
|||
|
median = \(x) median(x, na.rm = TRUE),
|
|||
|
n_miss = \(x) sum(is.na(x))
|
|||
|
)),
|
|||
|
n = n()
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 9
|
|||
|
#> a_median a_n_miss b_median b_n_miss c_median c_n_miss d_median d_n_miss n
|
|||
|
#> <dbl> <int> <dbl> <int> <dbl> <int> <dbl> <int> <int>
|
|||
|
#> 1 0.429 1 -0.721 1 -0.796 2 0.704 0 5</pre>
|
|||
|
</div>
|
|||
|
<p>If you look carefully, you might intuit that the columns are named using using a glue specification (<a href="#sec-glue" data-type="xref">#sec-glue</a>) like <code>{.col}_{.fn}</code> where <code>.col</code> is the name of the original column and <code>.fn</code> is the name of the function. That’s not a coincidence! As you’ll learn in the next section, you can use <code>.names</code> argument to supply your own glue spec.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="column-names" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Column names</h2>
|
|||
|
<p>The result of <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> is named according to the specification provided in the <code>.names</code> argument. We could specify our own if we wanted the name of the function to come first<span data-type="footnote">You can’t currently change the order of the columns, but you could reorder them after the fact using <code><a href="#chp-https://dplyr.tidyverse.org/reference/relocate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/relocate</a></code> or similar.</span>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
summarise(
|
|||
|
across(
|
|||
|
a:d,
|
|||
|
list(
|
|||
|
median = \(x) median(x, na.rm = TRUE),
|
|||
|
n_miss = \(x) sum(is.na(x))
|
|||
|
),
|
|||
|
.names = "{.fn}_{.col}"
|
|||
|
),
|
|||
|
n = n(),
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 9
|
|||
|
#> median_a n_miss_a median_b n_miss_b median_c n_miss_c median_d n_miss_d n
|
|||
|
#> <dbl> <int> <dbl> <int> <dbl> <int> <dbl> <int> <int>
|
|||
|
#> 1 0.429 1 -0.721 1 -0.796 2 0.704 0 5</pre>
|
|||
|
</div>
|
|||
|
<p>The <code>.names</code> argument is particularly important when you use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> with <code><a href="#chp-https://dplyr.tidyverse.org/reference/mutate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/mutate</a></code>. By default the output of <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> is given the same names as the inputs. This means that <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> inside of <code><a href="#chp-https://dplyr.tidyverse.org/reference/mutate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/mutate</a></code> will replace existing columns. For example, here we use <code><a href="#chp-https://dplyr.tidyverse.org/reference/coalesce" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/coalesce</a></code> to replace <code>NA</code>s with <code>0</code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
mutate(
|
|||
|
across(a:d, \(x) coalesce(x, 0))
|
|||
|
)
|
|||
|
#> # A tibble: 5 × 4
|
|||
|
#> a b c d
|
|||
|
#> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 0 -0.463 0 2.13
|
|||
|
#> 2 -0.382 -0.980 0 0.704
|
|||
|
#> 3 0.434 0 -1.06 0.715
|
|||
|
#> 4 1.06 1.21 -0.796 -1.09
|
|||
|
#> 5 0.424 -1.28 -0.785 0.402</pre>
|
|||
|
</div>
|
|||
|
<p>If you’d like to instead create new columns, you can use the <code>.names</code> argument to give the output new names:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |>
|
|||
|
mutate(
|
|||
|
across(a:d, \(x) abs(x), .names = "{.col}_abs")
|
|||
|
)
|
|||
|
#> # A tibble: 5 × 8
|
|||
|
#> a b c d a_abs b_abs c_abs d_abs
|
|||
|
#> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 NA -0.463 NA 2.13 NA 0.463 NA 2.13
|
|||
|
#> 2 -0.382 -0.980 NA 0.704 0.382 0.980 NA 0.704
|
|||
|
#> 3 0.434 NA -1.06 0.715 0.434 NA 1.06 0.715
|
|||
|
#> 4 1.06 1.21 -0.796 -1.09 1.06 1.21 0.796 1.09
|
|||
|
#> 5 0.424 -1.28 -0.785 0.402 0.424 1.28 0.785 0.402</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="filtering" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Filtering</h2>
|
|||
|
<p><code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> is a great match for <code><a href="#chp-https://dplyr.tidyverse.org/reference/summarise" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/summarise</a></code> and <code><a href="#chp-https://dplyr.tidyverse.org/reference/mutate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/mutate</a></code> but it’s more awkward to use with <code><a href="#chp-https://dplyr.tidyverse.org/reference/filter" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/filter</a></code>, because you usually combine multiple conditions with either <code>|</code> or <code>&</code>. It’s clear that <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> can help to create multiple logical columns, but then what? So dplyr provides two variants of <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> called <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> and <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_miss |> filter(is.na(a) | is.na(b) | is.na(c) | is.na(d))
|
|||
|
#> # A tibble: 3 × 4
|
|||
|
#> a b c d
|
|||
|
#> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 NA -0.463 NA 2.13
|
|||
|
#> 2 -0.382 -0.980 NA 0.704
|
|||
|
#> 3 0.434 NA -1.06 0.715
|
|||
|
# same as:
|
|||
|
df_miss |> filter(if_any(a:d, is.na))
|
|||
|
#> # A tibble: 3 × 4
|
|||
|
#> a b c d
|
|||
|
#> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 NA -0.463 NA 2.13
|
|||
|
#> 2 -0.382 -0.980 NA 0.704
|
|||
|
#> 3 0.434 NA -1.06 0.715
|
|||
|
|
|||
|
df_miss |> filter(is.na(a) & is.na(b) & is.na(c) & is.na(d))
|
|||
|
#> # A tibble: 0 × 4
|
|||
|
#> # … with 4 variables: a <dbl>, b <dbl>, c <dbl>, d <dbl>
|
|||
|
# same as:
|
|||
|
df_miss |> filter(if_all(a:d, is.na))
|
|||
|
#> # A tibble: 0 × 4
|
|||
|
#> # … with 4 variables: a <dbl>, b <dbl>, c <dbl>, d <dbl></pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="across-in-functions" data-type="sect2">
|
|||
|
<h2>
|
|||
|
<code>across()</code> in functions</h2>
|
|||
|
<p><code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> is particularly useful to program with because it allows you to operate on multiple columns. For example, <a href="#chp-https://twitter.com/_wurli/status/1571836746899283969" data-type="xref">#chp-https://twitter.com/_wurli/status/1571836746899283969</a> uses this little helper which wraps a bunch of lubridate function to expand all date columns into year, month, and day columns:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">library(lubridate)
|
|||
|
#> Loading required package: timechange
|
|||
|
#>
|
|||
|
#> Attaching package: 'lubridate'
|
|||
|
#> The following objects are masked from 'package:base':
|
|||
|
#>
|
|||
|
#> date, intersect, setdiff, union
|
|||
|
|
|||
|
expand_dates <- function(df) {
|
|||
|
df |>
|
|||
|
mutate(
|
|||
|
across(where(is.Date), list(year = year, month = month, day = mday))
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
df_date <- tibble(
|
|||
|
name = c("Amy", "Bob"),
|
|||
|
date = ymd(c("2009-08-03", "2010-01-16"))
|
|||
|
)
|
|||
|
|
|||
|
df_date |>
|
|||
|
expand_dates()
|
|||
|
#> # A tibble: 2 × 5
|
|||
|
#> name date date_year date_month date_day
|
|||
|
#> <chr> <date> <dbl> <dbl> <int>
|
|||
|
#> 1 Amy 2009-08-03 2009 8 3
|
|||
|
#> 2 Bob 2010-01-16 2010 1 16</pre>
|
|||
|
</div>
|
|||
|
<p><code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> also makes it easy to supply multiple columns in a single argument because the first argument uses tidy-select; you just need to remember to embrace that argument, as we discussed in <a href="#sec-embracing" data-type="xref">#sec-embracing</a>. For example, this function will compute the means of numeric columns by default. But by supplying the second argument you can choose to summarize just selected columns:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">summarise_means <- function(df, summary_vars = where(is.numeric)) {
|
|||
|
df |>
|
|||
|
summarise(
|
|||
|
across({{ summary_vars }}, \(x) mean(x, na.rm = TRUE)),
|
|||
|
n = n()
|
|||
|
)
|
|||
|
}
|
|||
|
diamonds |>
|
|||
|
group_by(clarity) |>
|
|||
|
summarise_means()
|
|||
|
#> # A tibble: 8 × 9
|
|||
|
#> clarity carat depth table price x y z n
|
|||
|
#> <ord> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <int>
|
|||
|
#> 1 I1 1.28 62.7 58.3 3924. 6.76 6.71 4.21 741
|
|||
|
#> 2 SI2 1.08 61.8 57.9 5063. 6.40 6.40 3.95 9194
|
|||
|
#> 3 SI1 0.850 61.9 57.7 3996. 5.89 5.89 3.64 13065
|
|||
|
#> 4 VS2 0.764 61.7 57.4 3925. 5.66 5.66 3.49 12258
|
|||
|
#> 5 VS1 0.727 61.7 57.3 3839. 5.57 5.58 3.44 8171
|
|||
|
#> 6 VVS2 0.596 61.7 57.0 3284. 5.22 5.23 3.22 5066
|
|||
|
#> # … with 2 more rows
|
|||
|
|
|||
|
diamonds |>
|
|||
|
group_by(clarity) |>
|
|||
|
summarise_means(c(carat, x:z))
|
|||
|
#> # A tibble: 8 × 6
|
|||
|
#> clarity carat x y z n
|
|||
|
#> <ord> <dbl> <dbl> <dbl> <dbl> <int>
|
|||
|
#> 1 I1 1.28 6.76 6.71 4.21 741
|
|||
|
#> 2 SI2 1.08 6.40 6.40 3.95 9194
|
|||
|
#> 3 SI1 0.850 5.89 5.89 3.64 13065
|
|||
|
#> 4 VS2 0.764 5.66 5.66 3.49 12258
|
|||
|
#> 5 VS1 0.727 5.57 5.58 3.44 8171
|
|||
|
#> 6 VVS2 0.596 5.22 5.23 3.22 5066
|
|||
|
#> # … with 2 more rows</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="vs-pivot_longer" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Vs<code>pivot_longer()</code>
|
|||
|
</h2>
|
|||
|
<p>Before we go on, it’s worth pointing out an interesting connection between <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> and <code><a href="#chp-https://tidyr.tidyverse.org/reference/pivot_longer" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/pivot_longer</a></code> (<a href="#sec-pivoting" data-type="xref">#sec-pivoting</a>). In many cases, you perform the same calculations by first pivoting the data and then performing the operations by group rather than by column. For example, take this multi-function summary:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df |>
|
|||
|
summarise(across(a:d, list(median = median, mean = mean)))
|
|||
|
#> # A tibble: 1 × 8
|
|||
|
#> a_median a_mean b_median b_mean c_median c_mean d_median d_mean
|
|||
|
#> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 0.0380 0.205 -0.0163 0.0910 0.260 0.0716 0.540 0.508</pre>
|
|||
|
</div>
|
|||
|
<p>We could compute the same values by pivoting longer and then summarizing:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">long <- df |>
|
|||
|
pivot_longer(a:d) |>
|
|||
|
group_by(name) |>
|
|||
|
summarise(
|
|||
|
median = median(value),
|
|||
|
mean = mean(value)
|
|||
|
)
|
|||
|
long
|
|||
|
#> # A tibble: 4 × 3
|
|||
|
#> name median mean
|
|||
|
#> <chr> <dbl> <dbl>
|
|||
|
#> 1 a 0.0380 0.205
|
|||
|
#> 2 b -0.0163 0.0910
|
|||
|
#> 3 c 0.260 0.0716
|
|||
|
#> 4 d 0.540 0.508</pre>
|
|||
|
</div>
|
|||
|
<p>And if you wanted the same structure as <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> you could pivot again:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">long |>
|
|||
|
pivot_wider(
|
|||
|
names_from = name,
|
|||
|
values_from = c(median, mean),
|
|||
|
names_vary = "slowest",
|
|||
|
names_glue = "{name}_{.value}"
|
|||
|
)
|
|||
|
#> # A tibble: 1 × 8
|
|||
|
#> a_median a_mean b_median b_mean c_median c_mean d_median d_mean
|
|||
|
#> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 0.0380 0.205 -0.0163 0.0910 0.260 0.0716 0.540 0.508</pre>
|
|||
|
</div>
|
|||
|
<p>This is a useful technique to know about because sometimes you’ll hit a problem that’s not currently possible to solve with <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>: when you have groups of columns that you want to compute with simultaneously. For example, imagine that our data frame contains both values and weights and we want to compute a weighted mean:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_paired <- tibble(
|
|||
|
a_val = rnorm(10),
|
|||
|
a_wts = runif(10),
|
|||
|
b_val = rnorm(10),
|
|||
|
b_wts = runif(10),
|
|||
|
c_val = rnorm(10),
|
|||
|
c_wts = runif(10),
|
|||
|
d_val = rnorm(10),
|
|||
|
d_wts = runif(10)
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>There’s currently no way to do this with <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code><span data-type="footnote">Maybe there will be one day, but currently we don’t see how.</span>, but it’s relatively straightforward with <code><a href="#chp-https://tidyr.tidyverse.org/reference/pivot_longer" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/pivot_longer</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_long <- df_paired |>
|
|||
|
pivot_longer(
|
|||
|
everything(),
|
|||
|
names_to = c("group", ".value"),
|
|||
|
names_sep = "_"
|
|||
|
)
|
|||
|
df_long
|
|||
|
#> # A tibble: 40 × 3
|
|||
|
#> group val wts
|
|||
|
#> <chr> <dbl> <dbl>
|
|||
|
#> 1 a 0.404 0.678
|
|||
|
#> 2 b 1.74 0.650
|
|||
|
#> 3 c -0.921 0.261
|
|||
|
#> 4 d -0.953 0.327
|
|||
|
#> 5 a 2.04 0.665
|
|||
|
#> 6 b -1.64 0.815
|
|||
|
#> # … with 34 more rows
|
|||
|
|
|||
|
df_long |>
|
|||
|
group_by(group) |>
|
|||
|
summarise(mean = weighted.mean(val, wts))
|
|||
|
#> # A tibble: 4 × 2
|
|||
|
#> group mean
|
|||
|
#> <chr> <dbl>
|
|||
|
#> 1 a 0.109
|
|||
|
#> 2 b 0.585
|
|||
|
#> 3 c -0.746
|
|||
|
#> 4 d -0.0142</pre>
|
|||
|
</div>
|
|||
|
<p>If needed, you could <code><a href="#chp-https://tidyr.tidyverse.org/reference/pivot_wider" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/pivot_wider</a></code> this back to the original form.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="exercises" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Exercises</h2>
|
|||
|
<ol type="1"><li><p>Compute the number of unique values in each column of <code><a href="#chp-https://allisonhorst.github.io/palmerpenguins/reference/penguins" data-type="xref">#chp-https://allisonhorst.github.io/palmerpenguins/reference/penguins</a></code>.</p></li>
|
|||
|
<li><p>Compute the mean of every column in <code>mtcars</code>.</p></li>
|
|||
|
<li><p>Group <code>diamonds</code> by <code>cut</code>, <code>clarity</code>, and <code>color</code> then count the number of observations and the mean of each numeric column.</p></li>
|
|||
|
<li><p>What happens if you use a list of functions, but don’t name them? How is the output named?</p></li>
|
|||
|
<li><p>It is possible to use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> inside <code><a href="#chp-https://dplyr.tidyverse.org/reference/filter" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/filter</a></code> where it’s equivalent to <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>. Can you explain why?</p></li>
|
|||
|
<li><p>Adjust <code>expand_dates()</code> to automatically remove the date columns after they’ve been expanded. Do you need to embrace any arguments?</p></li>
|
|||
|
<li>
|
|||
|
<p>Explain what each step of the pipeline in this function does. What special feature of <code>where()</code> are we taking advantage of?</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">show_missing <- function(df, group_vars, summary_vars = everything()) {
|
|||
|
df |>
|
|||
|
group_by(pick({{ group_vars }})) |>
|
|||
|
summarise(
|
|||
|
across({{ summary_vars }}, \(x) sum(is.na(x))),
|
|||
|
.groups = "drop"
|
|||
|
) |>
|
|||
|
select(where(\(x) any(x > 0)))
|
|||
|
}
|
|||
|
nycflights13::flights |> show_missing(c(year, month, day))</pre>
|
|||
|
</div>
|
|||
|
</li>
|
|||
|
</ol></section>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="reading-multiple-files" data-type="sect1">
|
|||
|
<h1>
|
|||
|
Reading multiple files</h1>
|
|||
|
<p>In the previous section, you learned how to use <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code> to repeat a transformation on multiple columns. In this section, you’ll learn how to use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> to do something to every file in a directory. Let’s start with a little motivation: imagine you have a directory full of excel spreadsheets<span data-type="footnote">If you instead had a directory of csv files with the same format, you can use the technique from <a href="#sec-readr-directory" data-type="xref">#sec-readr-directory</a>.</span> you want to read. You could do it with copy and paste:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">data2019 <- readxl::read_excel("data/y2019.xlsx")
|
|||
|
data2020 <- readxl::read_excel("data/y2020.xlsx")
|
|||
|
data2021 <- readxl::read_excel("data/y2021.xlsx")
|
|||
|
data2022 <- readxl::read_excel("data/y2022.xlsx")</pre>
|
|||
|
</div>
|
|||
|
<p>And then use <code><a href="#chp-https://dplyr.tidyverse.org/reference/bind_rows" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/bind_rows</a></code> to combine them all together:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">data <- bind_rows(data2019, data2020, data2021, data2022)</pre>
|
|||
|
</div>
|
|||
|
<p>You can imagine that this would get tedious quickly, especially if you had hundreds of files, not just four. The following sections show you how to automate this sort of task. There are three basic steps: use <code><a href="#chp-https://rdrr.io/r/base/list.files" data-type="xref">#chp-https://rdrr.io/r/base/list.files</a></code> to list all the files in a directory, then use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> to read each of them into a list, then use <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code> to combine them into a single data frame. We’ll then discuss how you can handle situations of increasing heterogeneity, where you can’t do exactly the same thing to every file.</p>
|
|||
|
|
|||
|
<section id="listing-files-in-a-directory" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Listing files in a directory</h2>
|
|||
|
<p>As the name suggests, <code><a href="#chp-https://rdrr.io/r/base/list.files" data-type="xref">#chp-https://rdrr.io/r/base/list.files</a></code> lists the files in a directory. TO CONSIDER: why not use it via the more obvious name <code><a href="#chp-https://rdrr.io/r/base/list.files" data-type="xref">#chp-https://rdrr.io/r/base/list.files</a></code>? You’ll almost always use three arguments:</p>
|
|||
|
<ul><li><p>The first argument, <code>path</code>, is the directory to look in.</p></li>
|
|||
|
<li><p><code>pattern</code> is a regular expression used to filter the file names. The most common pattern is something like <code>[.]xlsx$</code> or <code>[.]csv$</code> to find all files with a specified extension.</p></li>
|
|||
|
<li><p><code>full.names</code> determines whether or not the directory name should be included in the output. You almost always want this to be <code>TRUE</code>.</p></li>
|
|||
|
</ul><p>To make our motivating example concrete, this book contains a folder with 12 excel spreadsheets containing data from the gapminder package. Each file contains one year’s worth of data for 142 countries. We can list them all with the appropriate call to <code><a href="#chp-https://rdrr.io/r/base/list.files" data-type="xref">#chp-https://rdrr.io/r/base/list.files</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths <- list.files("data/gapminder", pattern = "[.]xlsx$", full.names = TRUE)
|
|||
|
paths
|
|||
|
#> [1] "data/gapminder/1952.xlsx" "data/gapminder/1957.xlsx"
|
|||
|
#> [3] "data/gapminder/1962.xlsx" "data/gapminder/1967.xlsx"
|
|||
|
#> [5] "data/gapminder/1972.xlsx" "data/gapminder/1977.xlsx"
|
|||
|
#> [7] "data/gapminder/1982.xlsx" "data/gapminder/1987.xlsx"
|
|||
|
#> [9] "data/gapminder/1992.xlsx" "data/gapminder/1997.xlsx"
|
|||
|
#> [11] "data/gapminder/2002.xlsx" "data/gapminder/2007.xlsx"</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="lists" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Lists</h2>
|
|||
|
<p>Now that we have these 12 paths, we could call <code>read_excel()</code> 12 times to get 12 data frames:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="r">gapminder_1952 <- readxl::read_excel("data/gapminder/1952.xlsx")
|
|||
|
gapminder_1957 <- readxl::read_excel("data/gapminder/1957.xlsx")
|
|||
|
gapminder_1962 <- readxl::read_excel("data/gapminder/1962.xlsx")
|
|||
|
...,
|
|||
|
gapminder_2007 <- readxl::read_excel("data/gapminder/2007.xlsx")</pre>
|
|||
|
</div>
|
|||
|
<p>But putting each sheet into its own variable is going to make it hard to work with them a few steps down the road. Instead, they’ll be easier to work with if we put them into a single object. A list is the perfect tool for this job:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- list(
|
|||
|
readxl::read_excel("data/gapminder/1952.xlsx"),
|
|||
|
readxl::read_excel("data/gapminder/1957.xlsx"),
|
|||
|
readxl::read_excel("data/gapminder/1962.xlsx"),
|
|||
|
...,
|
|||
|
readxl::read_excel("data/gapminder/2007.xlsx")
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>Now that you have these data frames in a list, how do you get one out? You can use <code>files[[i]]</code> to extract the i-th element:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files[[3]]
|
|||
|
#> # A tibble: 142 × 5
|
|||
|
#> country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 32.0 10267083 853.
|
|||
|
#> 2 Albania Europe 64.8 1728137 2313.
|
|||
|
#> 3 Algeria Africa 48.3 11000948 2551.
|
|||
|
#> 4 Angola Africa 34 4826015 4269.
|
|||
|
#> 5 Argentina Americas 65.1 21283783 7133.
|
|||
|
#> 6 Australia Oceania 70.9 10794968 12217.
|
|||
|
#> # … with 136 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>We’ll come back to <code>[[</code> in more detail in <a href="#sec-subset-one" data-type="xref">#sec-subset-one</a>.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="purrrmap-and-list_rbind" data-type="sect2">
|
|||
|
<h2>
|
|||
|
<code>purrr::map()</code> and <code>list_rbind()</code>
|
|||
|
</h2>
|
|||
|
<p>The code to collect those data frames in a list “by hand” is basically just as tedious to type as code that reads the files one-by-one. Happily, we can use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> to make even better use of our <code>paths</code> vector. <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> is similar to<code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>, but instead of doing something to each column in a data frame, it does something to each element of a vector.<code>map(x, f)</code> is shorthand for:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">list(
|
|||
|
f(x[[1]]),
|
|||
|
f(x[[2]]),
|
|||
|
...,
|
|||
|
f(x[[n]])
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>So we can use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> get a list of 12 data frames:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- map(paths, readxl::read_excel)
|
|||
|
length(files)
|
|||
|
#> [1] 12
|
|||
|
|
|||
|
files[[1]]
|
|||
|
#> # A tibble: 142 × 5
|
|||
|
#> country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 28.8 8425333 779.
|
|||
|
#> 2 Albania Europe 55.2 1282697 1601.
|
|||
|
#> 3 Algeria Africa 43.1 9279525 2449.
|
|||
|
#> 4 Angola Africa 30.0 4232095 3521.
|
|||
|
#> 5 Argentina Americas 62.5 17876956 5911.
|
|||
|
#> 6 Australia Oceania 69.1 8691212 10040.
|
|||
|
#> # … with 136 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>(This is another data structure that doesn’t display particularly compactly with <code><a href="#chp-https://rdrr.io/r/utils/str" data-type="xref">#chp-https://rdrr.io/r/utils/str</a></code> so you might want to load into RStudio and inspect it with <code><a href="#chp-https://rdrr.io/r/utils/View" data-type="xref">#chp-https://rdrr.io/r/utils/View</a></code>).</p>
|
|||
|
<p>Now we can use <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code> to combine that list of data frames into a single data frame:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">list_rbind(files)
|
|||
|
#> # A tibble: 1,704 × 5
|
|||
|
#> country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 28.8 8425333 779.
|
|||
|
#> 2 Albania Europe 55.2 1282697 1601.
|
|||
|
#> 3 Algeria Africa 43.1 9279525 2449.
|
|||
|
#> 4 Angola Africa 30.0 4232095 3521.
|
|||
|
#> 5 Argentina Americas 62.5 17876956 5911.
|
|||
|
#> 6 Australia Oceania 69.1 8691212 10040.
|
|||
|
#> # … with 1,698 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>Or we could do both steps at once in pipeline:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |>
|
|||
|
map(readxl::read_excel) |>
|
|||
|
list_rbind()</pre>
|
|||
|
</div>
|
|||
|
<p>What if we want to pass in extra arguments to <code>read_excel()</code>? We use the same technique that we used with <code><a href="#chp-https://dplyr.tidyverse.org/reference/across" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/across</a></code>. For example, it’s often useful to peak at the first few row of the data with <code>n_max = 1</code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |>
|
|||
|
map(\(path) readxl::read_excel(path, n_max = 1)) |>
|
|||
|
list_rbind()
|
|||
|
#> # A tibble: 12 × 5
|
|||
|
#> country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 28.8 8425333 779.
|
|||
|
#> 2 Afghanistan Asia 30.3 9240934 821.
|
|||
|
#> 3 Afghanistan Asia 32.0 10267083 853.
|
|||
|
#> 4 Afghanistan Asia 34.0 11537966 836.
|
|||
|
#> 5 Afghanistan Asia 36.1 13079460 740.
|
|||
|
#> 6 Afghanistan Asia 38.4 14880372 786.
|
|||
|
#> # … with 6 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>This makes it clear that something is missing: there’s no <code>year</code> column because that value is recorded in the path, not the individual files. We’ll tackle that problem next.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="sec-data-in-the-path" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Data in the path</h2>
|
|||
|
<p>Sometimes the name of the file is itself data. In this example, the file name contains the year, which is not otherwise recorded in the individual files. To get that column into the final data frame, we need to do two things.</p>
|
|||
|
<p>First, we name the vector of paths. The easiest way to do this is with the <code><a href="#chp-https://rlang.r-lib.org/reference/set_names" data-type="xref">#chp-https://rlang.r-lib.org/reference/set_names</a></code> function, which can take a function. Here we use <code><a href="#chp-https://rdrr.io/r/base/basename" data-type="xref">#chp-https://rdrr.io/r/base/basename</a></code> to extract just the file name from the full path:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |> set_names(basename)
|
|||
|
#> 1952.xlsx 1957.xlsx
|
|||
|
#> "data/gapminder/1952.xlsx" "data/gapminder/1957.xlsx"
|
|||
|
#> 1962.xlsx 1967.xlsx
|
|||
|
#> "data/gapminder/1962.xlsx" "data/gapminder/1967.xlsx"
|
|||
|
#> 1972.xlsx 1977.xlsx
|
|||
|
#> "data/gapminder/1972.xlsx" "data/gapminder/1977.xlsx"
|
|||
|
#> 1982.xlsx 1987.xlsx
|
|||
|
#> "data/gapminder/1982.xlsx" "data/gapminder/1987.xlsx"
|
|||
|
#> 1992.xlsx 1997.xlsx
|
|||
|
#> "data/gapminder/1992.xlsx" "data/gapminder/1997.xlsx"
|
|||
|
#> 2002.xlsx 2007.xlsx
|
|||
|
#> "data/gapminder/2002.xlsx" "data/gapminder/2007.xlsx"</pre>
|
|||
|
</div>
|
|||
|
<p>Those names are automatically carried along by all the map functions, so the list of data frames will have those same names:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- paths |>
|
|||
|
set_names(basename) |>
|
|||
|
map(readxl::read_excel)</pre>
|
|||
|
</div>
|
|||
|
<p>That makes this call to <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> shorthand for:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- list(
|
|||
|
"1952.xlsx" = readxl::read_excel("data/gapminder/1952.xlsx"),
|
|||
|
"1957.xlsx" = readxl::read_excel("data/gapminder/1957.xlsx"),
|
|||
|
"1962.xlsx" = readxl::read_excel("data/gapminder/1962.xlsx"),
|
|||
|
...,
|
|||
|
"2007.xlsx" = readxl::read_excel("data/gapminder/2007.xlsx")
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>You can also use <code>[[</code> to extract elements by name:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files[["1962.xlsx"]]
|
|||
|
#> # A tibble: 142 × 5
|
|||
|
#> country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 32.0 10267083 853.
|
|||
|
#> 2 Albania Europe 64.8 1728137 2313.
|
|||
|
#> 3 Algeria Africa 48.3 11000948 2551.
|
|||
|
#> 4 Angola Africa 34 4826015 4269.
|
|||
|
#> 5 Argentina Americas 65.1 21283783 7133.
|
|||
|
#> 6 Australia Oceania 70.9 10794968 12217.
|
|||
|
#> # … with 136 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>Then we use the <code>names_to</code> argument to <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code> to tell it to save the names into a new column called <code>year</code> then use <code><a href="#chp-https://readr.tidyverse.org/reference/parse_number" data-type="xref">#chp-https://readr.tidyverse.org/reference/parse_number</a></code> to extract the number from the string.</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |>
|
|||
|
set_names(basename) |>
|
|||
|
map(readxl::read_excel) |>
|
|||
|
list_rbind(names_to = "year") |>
|
|||
|
mutate(year = parse_number(year))
|
|||
|
#> # A tibble: 1,704 × 6
|
|||
|
#> year country continent lifeExp pop gdpPercap
|
|||
|
#> <dbl> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 1952 Afghanistan Asia 28.8 8425333 779.
|
|||
|
#> 2 1952 Albania Europe 55.2 1282697 1601.
|
|||
|
#> 3 1952 Algeria Africa 43.1 9279525 2449.
|
|||
|
#> 4 1952 Angola Africa 30.0 4232095 3521.
|
|||
|
#> 5 1952 Argentina Americas 62.5 17876956 5911.
|
|||
|
#> 6 1952 Australia Oceania 69.1 8691212 10040.
|
|||
|
#> # … with 1,698 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>In more complicated cases, there might be other variables stored in the directory name, or maybe the file name contains multiple bits of data. In that case, use <code><a href="#chp-https://rlang.r-lib.org/reference/set_names" data-type="xref">#chp-https://rlang.r-lib.org/reference/set_names</a></code> (without any arguments) to record the full path, and then use <code><a href="#chp-https://tidyr.tidyverse.org/reference/separate_wider_delim" data-type="xref">#chp-https://tidyr.tidyverse.org/reference/separate_wider_delim</a></code> and friends to turn them into useful columns.</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit"># NOTE: this chapter also depends on dev tidyr (in addition to dev purrr and dev dplyr)
|
|||
|
paths |>
|
|||
|
set_names() |>
|
|||
|
map(readxl::read_excel) |>
|
|||
|
list_rbind(names_to = "year") |>
|
|||
|
separate_wider_delim(year, delim = "/", names = c(NA, "dir", "file")) |>
|
|||
|
separate_wider_delim(file, delim = ".", names = c("file", "ext"))
|
|||
|
#> # A tibble: 1,704 × 8
|
|||
|
#> dir file ext country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <chr> <chr> <chr> <dbl> <dbl> <dbl>
|
|||
|
#> 1 gapminder 1952 xlsx Afghanistan Asia 28.8 8425333 779.
|
|||
|
#> 2 gapminder 1952 xlsx Albania Europe 55.2 1282697 1601.
|
|||
|
#> 3 gapminder 1952 xlsx Algeria Africa 43.1 9279525 2449.
|
|||
|
#> 4 gapminder 1952 xlsx Angola Africa 30.0 4232095 3521.
|
|||
|
#> 5 gapminder 1952 xlsx Argentina Americas 62.5 17876956 5911.
|
|||
|
#> 6 gapminder 1952 xlsx Australia Oceania 69.1 8691212 10040.
|
|||
|
#> # … with 1,698 more rows</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="save-your-work" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Save your work</h2>
|
|||
|
<p>Now that you’ve done all this hard work to get to a nice tidy data frame, it’s a great time to save your work:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">gapminder <- paths |>
|
|||
|
set_names(basename) |>
|
|||
|
map(readxl::read_excel) |>
|
|||
|
list_rbind(names_to = "year") |>
|
|||
|
mutate(year = parse_number(year))
|
|||
|
|
|||
|
write_csv(gapminder, "gapminder.csv")</pre>
|
|||
|
</div>
|
|||
|
<p>Now when you come back to this problem in the future, you can read in a single csv file.</p>
|
|||
|
<p>If you’re working in a project, we’d suggest calling the file that does this sort of data prep work something like <code>0-cleanup.R.</code> The <code>0</code> in the file name suggests that this should be run before anything else.</p>
|
|||
|
<p>If your input data files change over time, you might consider learning a tool like <a href="#chp-https://docs.ropensci.org/targets/" data-type="xref">#chp-https://docs.ropensci.org/targets/</a> to set up your data cleaning code to automatically re-run whenever one of the input files is modified.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="many-simple-iterations" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Many simple iterations</h2>
|
|||
|
<p>Here we’ve just loaded the data directly from disk, and were lucky enough to get a tidy dataset. In most cases, you’ll need to do some additional tidying, and you have two basic basic options: you can do one round of iteration with a complex function, or do a multiple rounds of iteration with simple functions. In our experience most folks reach first for one complex iteration, but you’re often better by doing multiple simple iterations.</p>
|
|||
|
<p>For example, imagine that you want to read in a bunch of files, filter out missing values, pivot, and then combine. One way to approach the problem is write a function that takes a file and does all those steps then call <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> once:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">process_file <- function(path) {
|
|||
|
df <- read_csv(path)
|
|||
|
|
|||
|
df |>
|
|||
|
filter(!is.na(id)) |>
|
|||
|
mutate(id = tolower(id)) |>
|
|||
|
pivot_longer(jan:dec, names_to = "month")
|
|||
|
}
|
|||
|
|
|||
|
paths |>
|
|||
|
map(process_file) |>
|
|||
|
list_rbind()</pre>
|
|||
|
</div>
|
|||
|
<p>Alternatively, you could perform each step of <code>process_file()</code> to every file:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |>
|
|||
|
map(read_csv) |>
|
|||
|
map(\(df) df |> filter(!is.na(id))) |>
|
|||
|
map(\(df) df |> mutate(id = tolower(id))) |>
|
|||
|
map(\(df) df |> pivot_longer(jan:dec, names_to = "month")) |>
|
|||
|
list_rbind()</pre>
|
|||
|
</div>
|
|||
|
<p>We recommend this approach because it stops you getting fixated on getting the first file right because moving on to the rest. By considering all of the data when doing tidying and cleaning, you’re more likely to think holistically and end up with a higher quality result.</p>
|
|||
|
<p>In this particular example, there’s another optimization you could make, by binding all the data frames together earlier. Then you can rely on regular dplyr behavior:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |>
|
|||
|
map(read_csv) |>
|
|||
|
list_rbind() |>
|
|||
|
filter(!is.na(id)) |>
|
|||
|
mutate(id = tolower(id)) |>
|
|||
|
pivot_longer(jan:dec, names_to = "month")</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="heterogeneous-data" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Heterogeneous data</h2>
|
|||
|
<p>Unfortunately sometimes it’s not possible to go from <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> straight to <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code> because the data frames are so heterogeneous that <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code> either fails or yields a data frame that’s not very useful. In that case, it’s still useful to start by loading all of the files:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- paths |>
|
|||
|
map(readxl::read_excel) </pre>
|
|||
|
</div>
|
|||
|
<p>Then a very useful strategy is to capture the structure of the data frames to data so that you can explore it using your data science skills. One way to do so is with this handy <code>df_types</code> function that returns a tibble with one row for each column:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">df_types <- function(df) {
|
|||
|
tibble(
|
|||
|
col_name = names(df),
|
|||
|
col_type = map_chr(df, vctrs::vec_ptype_full),
|
|||
|
n_miss = map_int(df, \(x) sum(is.na(x)))
|
|||
|
)
|
|||
|
}
|
|||
|
|
|||
|
df_types(starwars)
|
|||
|
#> # A tibble: 14 × 3
|
|||
|
#> col_name col_type n_miss
|
|||
|
#> <chr> <chr> <int>
|
|||
|
#> 1 name character 0
|
|||
|
#> 2 height integer 6
|
|||
|
#> 3 mass double 28
|
|||
|
#> 4 hair_color character 5
|
|||
|
#> 5 skin_color character 0
|
|||
|
#> 6 eye_color character 0
|
|||
|
#> # … with 8 more rows
|
|||
|
df_types(nycflights13::flights)
|
|||
|
#> # A tibble: 19 × 3
|
|||
|
#> col_name col_type n_miss
|
|||
|
#> <chr> <chr> <int>
|
|||
|
#> 1 year integer 0
|
|||
|
#> 2 month integer 0
|
|||
|
#> 3 day integer 0
|
|||
|
#> 4 dep_time integer 8255
|
|||
|
#> 5 sched_dep_time integer 0
|
|||
|
#> 6 dep_delay double 8255
|
|||
|
#> # … with 13 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>You can then apply this function all of the files, and maybe do some pivoting to make it easy to see where there are differences. For example, this makes it easy to verify that the gapminder spreadsheets that we’ve been working with are all quite homogeneous:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files |>
|
|||
|
map(df_types) |>
|
|||
|
list_rbind(names_to = "file_name") |>
|
|||
|
select(-n_miss) |>
|
|||
|
pivot_wider(names_from = col_name, values_from = col_type)
|
|||
|
#> # A tibble: 12 × 6
|
|||
|
#> file_name country continent lifeExp pop gdpPercap
|
|||
|
#> <chr> <chr> <chr> <chr> <chr> <chr>
|
|||
|
#> 1 1952.xlsx character character double double double
|
|||
|
#> 2 1957.xlsx character character double double double
|
|||
|
#> 3 1962.xlsx character character double double double
|
|||
|
#> 4 1967.xlsx character character double double double
|
|||
|
#> 5 1972.xlsx character character double double double
|
|||
|
#> 6 1977.xlsx character character double double double
|
|||
|
#> # … with 6 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>If the files have heterogeneous formats you might need to do more processing before you can successfully merge them. Unfortunately we’re now going to leave you to figure that out on your own, but you might want to read about <code><a href="#chp-https://purrr.tidyverse.org/reference/map_if" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map_if</a></code> and <code><a href="#chp-https://purrr.tidyverse.org/reference/map_if" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map_if</a></code>. <code><a href="#chp-https://purrr.tidyverse.org/reference/map_if" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map_if</a></code> allows you to selectively modify elements of a list based on their values; <code><a href="#chp-https://purrr.tidyverse.org/reference/map_if" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map_if</a></code> allows you to selectively modify elements based on their names.</p>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="handling-failures" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Handling failures</h2>
|
|||
|
<p>Sometimes the structure of your data might be sufficiently wild that you can’t even read all the files with a single command. And then you’ll encounter one of the downsides of map: it succeeds or fails as a whole. <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> will either successfully read all of the files in a directory or fail with an error, reading zero files. This is annoying: why does one failure prevent you from accessing all the other successes?</p>
|
|||
|
<p>Luckily, purrr comes with a helper to tackle this problem: <code><a href="#chp-https://purrr.tidyverse.org/reference/possibly" data-type="xref">#chp-https://purrr.tidyverse.org/reference/possibly</a></code>. <code><a href="#chp-https://purrr.tidyverse.org/reference/possibly" data-type="xref">#chp-https://purrr.tidyverse.org/reference/possibly</a></code> is what’s known as a function operator: it takes a function and returns a function with modified behavior. In particular, <code><a href="#chp-https://purrr.tidyverse.org/reference/possibly" data-type="xref">#chp-https://purrr.tidyverse.org/reference/possibly</a></code> changes a function from erroring to returning a value that you specify:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">files <- paths |>
|
|||
|
map(possibly(\(path) readxl::read_excel(path), NULL))
|
|||
|
|
|||
|
data <- files |> list_rbind()</pre>
|
|||
|
</div>
|
|||
|
<p>This works particularly well here because <code><a href="#chp-https://purrr.tidyverse.org/reference/list_c" data-type="xref">#chp-https://purrr.tidyverse.org/reference/list_c</a></code>, like many tidyverse functions, automatically ignores <code>NULL</code>s.</p>
|
|||
|
<p>Now you have all the data that can be read easily, and it’s time to tackle the hard part of figuring out why some files failed load and what do to about it. Start by getting the paths that failed:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">failed <- map_vec(files, is.null)
|
|||
|
paths[failed]
|
|||
|
#> character(0)</pre>
|
|||
|
</div>
|
|||
|
<p>Then call the import function again for each failure and figure out what went wrong.</p>
|
|||
|
</section>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="saving-multiple-outputs" data-type="sect1">
|
|||
|
<h1>
|
|||
|
Saving multiple outputs</h1>
|
|||
|
<p>In the last section, you learned about <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code>, which is useful for reading multiple files into a single object. In this section, we’ll now explore sort of the opposite problem: how can you take one or more R objects and save it to one or more files? We’ll explore this challenge using three examples:</p>
|
|||
|
<ul><li>Saving multiple data frames into one database.</li>
|
|||
|
<li>Saving multiple data frames into multiple csv files.</li>
|
|||
|
<li>Saving multiple plots to multiple <code>.png</code> files.</li>
|
|||
|
</ul>
|
|||
|
<section id="sec-save-database" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Writing to a database</h2>
|
|||
|
<p>Sometimes when working with many files at once, it’s not possible to fit all your data into memory at once, and you can’t do <code>map(files, read_csv)</code>. One approach to deal with this problem is to load your into a database so you can access just the bits you need with dbplyr.</p>
|
|||
|
<p>If you’re lucky, the database package you’re using will provide a handy function that takes a vector of paths and loads them all into the database. This is the case with duckdb’s <code>duckdb_read_csv()</code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">con <- DBI::dbConnect(duckdb::duckdb())
|
|||
|
duckdb::duckdb_read_csv(con, "gapminder", paths)</pre>
|
|||
|
</div>
|
|||
|
<p>This would work well here, but we don’t have csv files, instead we have excel spreadsheets. So we’re going to have to do it “by hand”. Learning to do it by hand will also help you when you have a bunch of csvs and the database that you’re working with doesn’t have one function that will load them all in.</p>
|
|||
|
<p>We need to start by creating a table that will fill in with data. The easiest way to do this is by creating a template, a dummy data frame that contains all the columns we want, but only a sampling of the data. For the gapminder data, we can make that template by reading a single file and adding the year to it:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">template <- readxl::read_excel(paths[[1]])
|
|||
|
template$year <- 1952
|
|||
|
template
|
|||
|
#> # A tibble: 142 × 6
|
|||
|
#> country continent lifeExp pop gdpPercap year
|
|||
|
#> <chr> <chr> <dbl> <dbl> <dbl> <dbl>
|
|||
|
#> 1 Afghanistan Asia 28.8 8425333 779. 1952
|
|||
|
#> 2 Albania Europe 55.2 1282697 1601. 1952
|
|||
|
#> 3 Algeria Africa 43.1 9279525 2449. 1952
|
|||
|
#> 4 Angola Africa 30.0 4232095 3521. 1952
|
|||
|
#> 5 Argentina Americas 62.5 17876956 5911. 1952
|
|||
|
#> 6 Australia Oceania 69.1 8691212 10040. 1952
|
|||
|
#> # … with 136 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>Now we can connect to the database, and use <code><a href="#chp-https://dbi.r-dbi.org/reference/dbCreateTable" data-type="xref">#chp-https://dbi.r-dbi.org/reference/dbCreateTable</a></code> to turn our template into database table:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">con <- DBI::dbConnect(duckdb::duckdb())
|
|||
|
DBI::dbCreateTable(con, "gapminder", template)</pre>
|
|||
|
</div>
|
|||
|
<p><code>dbCreateTable()</code> doesn’t use the data in <code>template</code>, just the variable names and types. So if we inspect the <code>gapminder</code> table now you’ll see that it’s empty but it has the variables we need with the types we expect:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">con |> tbl("gapminder")
|
|||
|
#> # Source: table<gapminder> [0 x 6]
|
|||
|
#> # Database: DuckDB 0.5.1 [root@Darwin 22.1.0:R 4.2.1/:memory:]
|
|||
|
#> # … with 6 variables: country <chr>, continent <chr>, lifeExp <dbl>, pop <dbl>,
|
|||
|
#> # gdpPercap <dbl>, year <dbl></pre>
|
|||
|
</div>
|
|||
|
<p>Next, we need a function that takes a single file path, reads it into R, and adds the result to the <code>gapminder</code> table. We can do that by combining <code>read_excel()</code> with <code><a href="#chp-https://dbi.r-dbi.org/reference/dbAppendTable" data-type="xref">#chp-https://dbi.r-dbi.org/reference/dbAppendTable</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">append_file <- function(path) {
|
|||
|
df <- readxl::read_excel(path)
|
|||
|
df$year <- parse_number(basename(path))
|
|||
|
|
|||
|
DBI::dbAppendTable(con, "gapminder", df)
|
|||
|
}</pre>
|
|||
|
</div>
|
|||
|
<p>Now we need to call <code>append_csv()</code> once for each element of <code>paths</code>. That’s certainly possible with <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |> map(append_file)</pre>
|
|||
|
</div>
|
|||
|
<p>But we don’t care about the output of <code>append_file()</code>, so instead of <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> it’s slightly nicer to use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code>. <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> does exactly the same thing as <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> but throws the output away:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">paths |> walk(append_file)</pre>
|
|||
|
</div>
|
|||
|
<p>Now we can see if we have all the data in our table:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">con |>
|
|||
|
tbl("gapminder") |>
|
|||
|
count(year)
|
|||
|
#> # Source: SQL [?? x 2]
|
|||
|
#> # Database: DuckDB 0.5.1 [root@Darwin 22.1.0:R 4.2.1/:memory:]
|
|||
|
#> year n
|
|||
|
#> <dbl> <dbl>
|
|||
|
#> 1 1952 142
|
|||
|
#> 2 1987 142
|
|||
|
#> 3 1957 142
|
|||
|
#> 4 1992 142
|
|||
|
#> 5 1962 142
|
|||
|
#> 6 1997 142
|
|||
|
#> # … with more rows</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="writing-csv-files" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Writing csv files</h2>
|
|||
|
<p>The same basic principle applies if we want to write multiple csv files, one for each group. Let’s imagine that we want to take the <code><a href="#chp-https://ggplot2.tidyverse.org/reference/diamonds" data-type="xref">#chp-https://ggplot2.tidyverse.org/reference/diamonds</a></code> data and save one csv file for each <code>clarity</code>. First we need to make those individual datasets. There are many ways you could do that, but there’s one way we particularly like: <code><a href="#chp-https://dplyr.tidyverse.org/reference/group_nest" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/group_nest</a></code>.</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">by_clarity <- diamonds |>
|
|||
|
group_nest(clarity)
|
|||
|
|
|||
|
by_clarity
|
|||
|
#> # A tibble: 8 × 2
|
|||
|
#> clarity data
|
|||
|
#> <ord> <list<tibble[,9]>>
|
|||
|
#> 1 I1 [741 × 9]
|
|||
|
#> 2 SI2 [9,194 × 9]
|
|||
|
#> 3 SI1 [13,065 × 9]
|
|||
|
#> 4 VS2 [12,258 × 9]
|
|||
|
#> 5 VS1 [8,171 × 9]
|
|||
|
#> 6 VVS2 [5,066 × 9]
|
|||
|
#> # … with 2 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>This gives us a new tibble with eight rows and two columns. <code>clarity</code> is our grouping variable and <code>data</code> is a list-column containing one tibble for each unique value of <code>clarity</code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">by_clarity$data[[1]]
|
|||
|
#> # A tibble: 741 × 9
|
|||
|
#> carat cut color depth table price x y z
|
|||
|
#> <dbl> <ord> <ord> <dbl> <dbl> <int> <dbl> <dbl> <dbl>
|
|||
|
#> 1 0.32 Premium E 60.9 58 345 4.38 4.42 2.68
|
|||
|
#> 2 1.17 Very Good J 60.2 61 2774 6.83 6.9 4.13
|
|||
|
#> 3 1.01 Premium F 61.8 60 2781 6.39 6.36 3.94
|
|||
|
#> 4 1.01 Fair E 64.5 58 2788 6.29 6.21 4.03
|
|||
|
#> 5 0.96 Ideal F 60.7 55 2801 6.37 6.41 3.88
|
|||
|
#> 6 1.04 Premium G 62.2 58 2801 6.46 6.41 4
|
|||
|
#> # … with 735 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>While we’re here, lets create a column that gives the name of output file, using <code><a href="#chp-https://dplyr.tidyverse.org/reference/mutate" data-type="xref">#chp-https://dplyr.tidyverse.org/reference/mutate</a></code> and <code><a href="#chp-https://stringr.tidyverse.org/reference/str_glue" data-type="xref">#chp-https://stringr.tidyverse.org/reference/str_glue</a></code>:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">by_clarity <- by_clarity |>
|
|||
|
mutate(path = str_glue("diamonds-{clarity}.csv"))
|
|||
|
|
|||
|
by_clarity
|
|||
|
#> # A tibble: 8 × 3
|
|||
|
#> clarity data path
|
|||
|
#> <ord> <list<tibble[,9]>> <glue>
|
|||
|
#> 1 I1 [741 × 9] diamonds-I1.csv
|
|||
|
#> 2 SI2 [9,194 × 9] diamonds-SI2.csv
|
|||
|
#> 3 SI1 [13,065 × 9] diamonds-SI1.csv
|
|||
|
#> 4 VS2 [12,258 × 9] diamonds-VS2.csv
|
|||
|
#> 5 VS1 [8,171 × 9] diamonds-VS1.csv
|
|||
|
#> 6 VVS2 [5,066 × 9] diamonds-VVS2.csv
|
|||
|
#> # … with 2 more rows</pre>
|
|||
|
</div>
|
|||
|
<p>So if we were going to save these data frames by hand, we might write something like:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">write_csv(by_clarity$data[[1]], by_clarity$path[[1]])
|
|||
|
write_csv(by_clarity$data[[2]], by_clarity$path[[2]])
|
|||
|
write_csv(by_clarity$data[[3]], by_clarity$path[[3]])
|
|||
|
...
|
|||
|
write_csv(by_clarity$by_clarity[[8]], by_clarity$path[[8]])</pre>
|
|||
|
</div>
|
|||
|
<p>This is a little different to our previous uses of <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> because there are two arguments that are changing, not just one. That means we need a new function: <code><a href="#chp-https://purrr.tidyverse.org/reference/map2" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map2</a></code>, which varies both the first and second arguments. And because we again don’t care about the output, we want <code><a href="#chp-https://purrr.tidyverse.org/reference/map2" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map2</a></code> rather than <code><a href="#chp-https://purrr.tidyverse.org/reference/map2" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map2</a></code>. That gives us:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">walk2(by_clarity$data, by_clarity$path, write_csv)</pre>
|
|||
|
</div>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="saving-plots" data-type="sect2">
|
|||
|
<h2>
|
|||
|
Saving plots</h2>
|
|||
|
<p>We can take the same basic approach to create many plots. Let’s first make a function that draws the plot we want:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">carat_histogram <- function(df) {
|
|||
|
ggplot(df, aes(carat)) + geom_histogram(binwidth = 0.1)
|
|||
|
}
|
|||
|
|
|||
|
carat_histogram(by_clarity$data[[1]])</pre>
|
|||
|
<div class="cell-output-display">
|
|||
|
<p><img src="iteration_files/figure-html/unnamed-chunk-70-1.png" class="img-fluid" width="576"/></p>
|
|||
|
</div>
|
|||
|
</div>
|
|||
|
<p>Now we can use <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> to create a list of many plots<span data-type="footnote">You can print <code>by_clarity$plot</code> to get a crude animation — you’ll get one plot for each element of <code>plots</code>. NOTE: this didn’t happen for me.</span> and their eventual file paths:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">by_clarity <- by_clarity |>
|
|||
|
mutate(
|
|||
|
plot = map(data, carat_histogram),
|
|||
|
path = str_glue("clarity-{clarity}.png")
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>Then use <code><a href="#chp-https://purrr.tidyverse.org/reference/map2" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map2</a></code> with <code><a href="#chp-https://ggplot2.tidyverse.org/reference/ggsave" data-type="xref">#chp-https://ggplot2.tidyverse.org/reference/ggsave</a></code> to save each plot:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">walk2(
|
|||
|
by_clarity$path,
|
|||
|
by_clarity$plot,
|
|||
|
\(path, plot) ggsave(path, plot, width = 6, height = 6)
|
|||
|
)</pre>
|
|||
|
</div>
|
|||
|
<p>This is shorthand for:</p>
|
|||
|
<div class="cell">
|
|||
|
<pre data-type="programlisting" data-code-language="downlit">ggsave(by_clarity$path[[1]], by_clarity$plot[[1]], width = 6, height = 6)
|
|||
|
ggsave(by_clarity$path[[2]], by_clarity$plot[[2]], width = 6, height = 6)
|
|||
|
ggsave(by_clarity$path[[3]], by_clarity$plot[[3]], width = 6, height = 6)
|
|||
|
...
|
|||
|
ggsave(by_clarity$path[[8]], by_clarity$plot[[8]], width = 6, height = 6)</pre>
|
|||
|
</div>
|
|||
|
|
|||
|
<!--
|
|||
|
### Exercises
|
|||
|
|
|||
|
1. Imagine you have a table of student data containing (amongst other variables) `school_name` and `student_id`. Sketch out what code you'd write if you want to save all the information for each student in file called `{student_id}.csv` in the `{school}` directory.
|
|||
|
-->
|
|||
|
</section>
|
|||
|
</section>
|
|||
|
|
|||
|
<section id="summary" data-type="sect1">
|
|||
|
<h1>
|
|||
|
Summary</h1>
|
|||
|
<p>In this chapter you’ve seen how to use explicit iteration to solve three problems that come up frequently when doing data science: manipulating multiple columns, reading multiple files, and saving multiple outputs. But in general, iteration is a super power: if you know the right iteration technique, you can easily go from fixing one problem to fixing all the problems. Once you’ve mastered the techniques in this chapter, we highly recommend learning more by reading the <a href="#chp-https://adv-r.hadley.nz/functionals" data-type="xref">#chp-https://adv-r.hadley.nz/functionals</a> of <em>Advanced R</em> and consulting the <a href="#chp-https://purrr.tidyverse" data-type="xref">#chp-https://purrr.tidyverse</a>.</p>
|
|||
|
<p>If you know much about iteration in other languages you might be surprised that we didn’t discuss the <code>for</code> loop. That’s because R’s orientation towards data analysis changes how we iterate: in most cases you can rely on an existing idiom to do something to each columns or each group. And when you can’t, you can often use a functional programming tool like <code><a href="#chp-https://purrr.tidyverse.org/reference/map" data-type="xref">#chp-https://purrr.tidyverse.org/reference/map</a></code> that does something to each element of a list. However, you will see <code>for</code> loops in wild-caught code, so you’ll learn about them in the next chapter where we’ll discuss some important base R tools.</p>
|
|||
|
|
|||
|
|
|||
|
</section>
|
|||
|
</section>
|