r4ds/workflow-style.Rmd

203 lines
8.0 KiB
Plaintext
Raw Normal View History

# Workflow: code style {#workflow-style}
```{r, results = "asis", echo = FALSE}
2022-02-22 05:54:38 +08:00
status("polishing")
```
Good coding style is like correct punctuation: you can manage without it, butitsuremakesthingseasiertoread.
2022-02-17 05:51:33 +08:00
Even as a very new programmer it's a good idea to work on your code style.
Use a consistent style makes it easier for others (including future-you!) to read your work, and is particularly important if you need to get help from someone else.
2022-02-22 05:39:02 +08:00
This chapter will introduce to the most important points of the [tidyverse style guide](https://style.tidyverse.org), which is used throughout this book.
Styling your code will feel a bit tedious to start with, but if you practice it, it will soon become second nature.
Additionally, there are some great tools to quickly restyle existing code, like the [styler](http://styler.r-lib.org) package by Lorenz Walthert.
Once you've installed it with `install.packages("styler")`, an easy way to use it is via RStudio's **command palette**.
The command palette lets you use any build-in RStudio command, as well as many addins provided by packages.
Open the palette by pressing Cmd/Ctrl + Shift + P, then type "styler" to see all the shortcuts provided by styler.
Figure \@ref(fig:styler) shows the results.
```{r styler}
#| echo: false
#| out.width: NULL
#| fig.cap: >
#| RStudio's command palette makes it easy to access every RStudio command
#| using only the keyboard.
#| fig.alt: >
#| A screenshot showing the command palette after typing "styler", showing
#| the four styling tool provided by the package.
knitr::include_graphics("screenshots/rstudio-palette.png")
```
2022-02-23 07:49:31 +08:00
```{r setup}
library(tidyverse)
library(nycflights13)
```
2022-02-22 05:39:02 +08:00
## Names
2022-02-17 21:09:09 +08:00
2022-03-03 23:31:46 +08:00
We talked briefly about names in Section \@ref(whats-in-a-name).
Remember that variable names (those created by `<-` and those created by `mutate()`) should use only lowercase letters, numbers, and `_`.
Use `_` to separate words within a name.
2022-02-22 05:39:02 +08:00
```{r, eval = FALSE}
# Strive for:
2022-02-23 07:49:31 +08:00
short_flights <- flights |> filter(air_time < 60)
2022-02-17 05:51:33 +08:00
2022-02-22 05:39:02 +08:00
# Avoid:
2022-03-03 23:31:46 +08:00
SHORTFLIGHTS <- flights |> filter(air_time < 60)
2022-02-22 05:39:02 +08:00
```
2022-02-17 05:51:33 +08:00
2022-02-22 05:39:02 +08:00
As a general rule of thumb, it's better to prefer long, descriptive names that are easy to understand, rather than concise names that are fast to type.
Short names save relatively little time when writing code (especially since autocomplete will help you finish typing them), but can be expensive when you come back to old need and need to puzzle out what a cryptic abbreviation means.
2022-03-03 23:31:46 +08:00
If you have a bunch of names for related things, do your best to be consistent.
It's easy for inconsistencies to arise when you forget a previous convention, so don't feel bad if you have to go back and rename things.
If you have a bunch of variables that are some variation on a theme you're generally better off giving them a common prefix, rather than a common suffix, because autocomplete works best on the start of a variable.
2022-02-23 07:49:31 +08:00
2022-02-22 05:39:02 +08:00
## Spaces
2022-03-03 23:31:46 +08:00
Put spaces on either side of mathematical operators apart from `^` (`+`, `-`, `==`, `<`, ...), and around the assignment operator (`<-`).
2022-02-17 21:09:09 +08:00
Don't put spaces inside or outside parentheses for regular function calls.
Always put a space after a comma, just like in regular English.
2022-02-22 05:39:02 +08:00
```{r, eval = FALSE}
# Strive for
(a + b)^2 / d
mean(x, na.rm = TRUE)
2022-02-22 05:39:02 +08:00
# Avoid
( a + b ) ^ 2/d
mean (x ,na.rm=TRUE)
```
2022-02-23 07:49:31 +08:00
It's OK to add extra spaces if it improves alignment.
For example, if you're creating multiple variables in `mutate()`, you might want to add spaces so that all the `=` line up.
2022-02-22 05:39:02 +08:00
```{r, eval = FALSE}
flights |>
mutate(
speed = air_time / distance,
dep_hour = dep_time %/% 100,
2022-02-23 07:49:31 +08:00
dep_minute = dep_time %% 100
2022-02-22 05:39:02 +08:00
)
```
## Pipes
2022-03-03 23:31:46 +08:00
`|>` should always have a space before it and should typically be followed by a newline.
If the function has named arguments (like `mutate()` or `summarise()`), always put each argument on a new line.
If the function doesn't have named arguments (like `select()` or `filter()` keep everything on one line unless it doesn't fit, in which case you should put each argument on its own line.
2022-02-23 07:49:31 +08:00
2022-03-03 23:31:46 +08:00
After the first step of the pipeline, indent each line by two spaces.
If you're putting arguments on their own line, indent each argument by an extra two spaces.
Make sure `)` is on its own line, and un-indented to match the horizontal position of the function name.
2022-02-17 21:09:09 +08:00
```{r, eval = FALSE}
2022-02-22 05:39:02 +08:00
# Strive for
flights |>
filter(!is.na(arr_delay), !is.na(tailnum)) |>
group_by(tailnum) |>
summarise(
delay = mean(arr_delay, na.rm = TRUE),
n = n()
)
# Avoid
flights|> filter(!is.na(arr_delay), !is.na(tailnum)) |>
group_by(tailnum) |> summarise(delay = mean(arr_delay, na.rm = TRUE),
n = n())
```
2022-03-03 23:31:46 +08:00
This structure makes it easier to add new steps, rearrange existing steps, modify elements within a step, and to get a 50,000 view by skimming the left-hand side.
It's OK to shirk some of these rules if your pipeline fits easily on one line.
But it's common for short snippets to grow longer, so you'll usually save time in the long run by starting with all the vertical space you need.
2022-02-22 05:39:02 +08:00
```{r, eval = FALSE}
# This fits compactly on one line
2022-02-17 21:09:09 +08:00
df |> mutate(y = x + 1)
2022-02-22 05:39:02 +08:00
2022-03-03 23:31:46 +08:00
# While this takes up 4x as many lines, it's easily extended to
# more variables and more steps in the future
2022-02-17 21:09:09 +08:00
df |>
mutate(
y = x + 1
)
```
2022-02-17 05:51:33 +08:00
2022-02-17 21:09:09 +08:00
The same basic rules apply to ggplot2, just treat `+` the same way as `|>`.
2022-02-17 21:09:09 +08:00
```{r, eval = FALSE}
2022-02-22 05:39:02 +08:00
flights |>
group_by(month) |>
2022-02-23 07:49:31 +08:00
summarise(
delay = mean(arr_delay, na.rm = TRUE)
) |>
2022-02-22 05:39:02 +08:00
ggplot(aes(month, delay)) +
geom_point() +
geom_line()
2022-02-17 21:09:09 +08:00
```
2022-03-03 23:31:46 +08:00
Again, if you can fit all of the arguments to a function on to a single line, put each argument on its own line:
2022-02-23 07:49:31 +08:00
```{r, eval = FALSE}
flights |>
group_by(dest) |>
summarise(
distance = mean(distance),
speed = mean(air_time / distance, na.rm = TRUE)
) |>
ggplot(aes(distance, speed)) +
geom_smooth(
method = "loess",
span = 0.5,
se = FALSE,
colour = "white",
size = 4
) +
geom_point()
```
2022-02-17 21:09:09 +08:00
Be wary of writing very long pipes, say longer than 10-15 lines.
2022-02-22 05:39:02 +08:00
Try to break them up into smaller sub-tasks, giving each task an informative name.
The names will help cue the reader into what's happening and makes it easier to check that intermediate results are as expected.
2022-02-17 21:09:09 +08:00
Whenever you can give something an informative name, you should give it an informative name.
Don't expect to get it right the first time!
This means breaking up long pipelines if there are intermediate states that can get good names.
2022-02-17 21:09:09 +08:00
## Organisation
2022-02-23 07:49:31 +08:00
Use comments to explain the "why" of your code, not the "how" or the "what".
2022-02-22 05:39:02 +08:00
If you simply describe what your code is doing in prose, you'll have to be careful to update the comment and code in tandem: if you change the code and forget to update the comment, they'll be inconsistent which will lead to confusion when you come back to your code in the future.
For data analysis code, use comments to explain your overall plan of attack and record important insight as you encounter them.
There's way to re-capture this knowledge from the code itself.
As your scripts get longer, use **sectioning** comments to break up your file into manageable pieces:
```{r, eval = FALSE}
# Load data --------------------------------------
# Plot data --------------------------------------
```
RStudio provides a keyboard shortcut to create these headers (Cmd/Ctrl + Shift + R), and will display them in the code navigation drop-down at the bottom-left of the editor, as shown in Figure \@ref(fig:rstudio-sections).
2022-02-22 05:39:02 +08:00
```{r rstudio-sections, echo = FALSE, out.width = NULL}
#| echo: false
#| out.width: NULL
#| fig.cap: >
#| After adding sectioning comments to your script, you can
#| easily navigate to them using the code navigation tool in the
#| bottom-left of the script editor.
knitr::include_graphics("screenshots/rstudio-nav.png")
```
2022-03-03 23:31:46 +08:00
## Exercises
1. Restyle each of the following pipelines following the guidelines above.
```{r, eval = FALSE}
flights|>filter(dest=="IAH")|>group_by(year,month,day)|>summarise(n=n(),delay=mean(arr_delay,na.rm=TRUE))|>filter(n>10)
```