Practice transforming and visualizing factors

library(tidyverse)
library(rcfss)
theme_set(theme_minimal())

Run the code below in your console to download this exercise as a set of R scripts.

usethis::use_course("uc-cfss/data-wrangling-relational-data-and-factors")
# load the data
data("gun_deaths")
gun_deaths
## # A tibble: 100,798 × 10
##       id  year month intent       police sex     age race        place education
##    <dbl> <dbl> <chr> <chr>         <dbl> <chr> <dbl> <chr>       <chr> <fct>    
##  1     1  2012 Jan   Suicide           0 M        34 Asian/Paci… Home  BA+      
##  2     2  2012 Jan   Suicide           0 F        21 White       Stre… Some col…
##  3     3  2012 Jan   Suicide           0 M        60 White       Othe… BA+      
##  4     4  2012 Feb   Suicide           0 M        64 White       Home  BA+      
##  5     5  2012 Feb   Suicide           0 M        31 White       Othe… HS/GED   
##  6     6  2012 Feb   Suicide           0 M        17 Native Ame… Home  Less tha…
##  7     7  2012 Feb   Undetermined      0 M        48 White       Home  HS/GED   
##  8     8  2012 Mar   Suicide           0 M        41 Native Ame… Home  HS/GED   
##  9     9  2012 Feb   Accidental        0 M        50 White       Othe… Some col…
## 10    10  2012 Feb   Suicide           0 M        NA Black       Home  <NA>     
## # … with 100,788 more rows

Convert month into a factor column

Click for the solution

# create a character vector with all month values
month_levels <- c(
  "Jan", "Feb", "Mar", "Apr", "May", "Jun",
  "Jul", "Aug", "Sep", "Oct", "Nov", "Dec"
)

# or use the built-in constant
month.abb
##  [1] "Jan" "Feb" "Mar" "Apr" "May" "Jun" "Jul" "Aug" "Sep" "Oct" "Nov" "Dec"
# use mutate() and factor() to convert the column and store the result
(gun_deaths <- gun_deaths %>%
  mutate(month = factor(month,
    levels = month_levels
  )))
## # A tibble: 100,798 × 10
##       id  year month intent       police sex     age race        place education
##    <dbl> <dbl> <fct> <chr>         <dbl> <chr> <dbl> <chr>       <chr> <fct>    
##  1     1  2012 Jan   Suicide           0 M        34 Asian/Paci… Home  BA+      
##  2     2  2012 Jan   Suicide           0 F        21 White       Stre… Some col…
##  3     3  2012 Jan   Suicide           0 M        60 White       Othe… BA+      
##  4     4  2012 Feb   Suicide           0 M        64 White       Home  BA+      
##  5     5  2012 Feb   Suicide           0 M        31 White       Othe… HS/GED   
##  6     6  2012 Feb   Suicide           0 M        17 Native Ame… Home  Less tha…
##  7     7  2012 Feb   Undetermined      0 M        48 White       Home  HS/GED   
##  8     8  2012 Mar   Suicide           0 M        41 Native Ame… Home  HS/GED   
##  9     9  2012 Feb   Accidental        0 M        50 White       Othe… Some col…
## 10    10  2012 Feb   Suicide           0 M        NA Black       Home  <NA>     
## # … with 100,788 more rows

Visualize the total gun deaths per month, in chronological order

Click for the solution

ggplot(
  data = gun_deaths,
  mapping = aes(x = month)
) +
  geom_bar() +
  labs(
    title = "Gun Deaths in the United States (2012-2014)",
    x = "Month",
    y = "Number of gun deaths"
  )

Visualize the total gun deaths per month, sorted from lowest to highest

Click for the solution

# with geom_col() and fct_reorder()
gun_deaths %>%
  count(month) %>%
  mutate(month = fct_reorder(.f = month, .x = n)) %>%
  ggplot(mapping = aes(x = month, y = n)) +
  geom_col() +
  labs(
    title = "Gun Deaths in the United States (2012-2014)",
    x = "Month",
    y = "Number of gun deaths"
  )

# with geom_bar() and fct_infreq()
gun_deaths %>%
  mutate(month = fct_infreq(f = month) %>%
    fct_rev()) %>%
  ggplot(mapping = aes(x = month)) +
  geom_bar() +
  labs(
    title = "Gun Deaths in the United States (2012-2014)",
    x = "Month",
    y = "Number of gun deaths"
  )

Visualize the frequency of intent of gun deaths using a bar chart, sorted from most to least frequent

Click for the solution

# identify all possible types of intent
intent_levels <- c("Accidental", "Homicide", "Suicide", "Undetermined")

gun_deaths %>%
  # remove rows with missing intent values
  drop_na(intent) %>%
  # parse_factor() is a tidyverse friendly form of factor()
  # ensure values are properly ordered from highest to lowest frequency
  mutate(intent = parse_factor(intent, levels = intent_levels) %>%
    fct_infreq() %>%
    fct_rev()) %>%
  ggplot(mapping = aes(x = intent)) +
  geom_bar() +
  labs(
    title = "Gun Deaths in the United States (2012-2014)",
    x = "Intent of death",
    y = "Number of gun deaths"
  ) +
  coord_flip()

Visualize total gun deaths by season of the year using a bar chart.

Hint: do not use cut() to create the season column.

Click for the solution

gun_deaths %>%
  # use fct_collapse() to condense into 4 categories
  mutate(season = fct_collapse(month,
    "Winter" = c("Jan", "Feb", "Mar"),
    "Spring" = c("Apr", "May", "Jun"),
    "Summer" = c("Jul", "Aug", "Sep"),
    "Fall" = c("Oct", "Nov", "Dec")
  )) %>%
  ggplot(mapping = aes(x = season)) +
  geom_bar() +
  labs(
    title = "Gun Deaths in the United States (2012-2014)",
    x = "Season",
    y = "Number of gun deaths"
  )

Session Info

devtools::session_info()
## ─ Session info ───────────────────────────────────────────────────────────────
##  setting  value
##  version  R version 4.2.0 (2022-04-22)
##  os       macOS Monterey 12.2.1
##  system   aarch64, darwin20
##  ui       X11
##  language (EN)
##  collate  en_US.UTF-8
##  ctype    en_US.UTF-8
##  tz       America/Chicago
##  date     2022-06-08
##  pandoc   2.17.1.1 @ /Applications/RStudio.app/Contents/MacOS/quarto/bin/ (via rmarkdown)
## 
## ─ Packages ───────────────────────────────────────────────────────────────────
##  package        * version date (UTC) lib source
##  assertthat       0.2.1   2019-03-21 [1] CRAN (R 4.2.0)
##  backports        1.4.1   2021-12-13 [1] CRAN (R 4.2.0)
##  blogdown         1.10    2022-05-10 [1] CRAN (R 4.2.0)
##  bookdown         0.26    2022-04-15 [1] CRAN (R 4.2.0)
##  brio             1.1.3   2021-11-30 [1] CRAN (R 4.2.0)
##  broom            0.8.0   2022-04-13 [1] CRAN (R 4.2.0)
##  bslib            0.3.1   2021-10-06 [1] CRAN (R 4.2.0)
##  cachem           1.0.6   2021-08-19 [1] CRAN (R 4.2.0)
##  callr            3.7.0   2021-04-20 [1] CRAN (R 4.2.0)
##  cellranger       1.1.0   2016-07-27 [1] CRAN (R 4.2.0)
##  cli              3.3.0   2022-04-25 [1] CRAN (R 4.2.0)
##  codetools        0.2-18  2020-11-04 [1] CRAN (R 4.2.0)
##  colorspace       2.0-3   2022-02-21 [1] CRAN (R 4.2.0)
##  crayon           1.5.1   2022-03-26 [1] CRAN (R 4.2.0)
##  DBI              1.1.2   2021-12-20 [1] CRAN (R 4.2.0)
##  dbplyr           2.2.0   2022-06-05 [1] CRAN (R 4.2.0)
##  desc             1.4.1   2022-03-06 [1] CRAN (R 4.2.0)
##  devtools         2.4.3   2021-11-30 [1] CRAN (R 4.2.0)
##  digest           0.6.29  2021-12-01 [1] CRAN (R 4.2.0)
##  dplyr          * 1.0.9   2022-04-28 [1] CRAN (R 4.2.0)
##  ellipsis         0.3.2   2021-04-29 [1] CRAN (R 4.2.0)
##  evaluate         0.15    2022-02-18 [1] CRAN (R 4.2.0)
##  fansi            1.0.3   2022-03-24 [1] CRAN (R 4.2.0)
##  farver           2.1.0   2021-02-28 [1] CRAN (R 4.2.0)
##  fastmap          1.1.0   2021-01-25 [1] CRAN (R 4.2.0)
##  forcats        * 0.5.1   2021-01-27 [1] CRAN (R 4.2.0)
##  fs               1.5.2   2021-12-08 [1] CRAN (R 4.2.0)
##  generics         0.1.2   2022-01-31 [1] CRAN (R 4.2.0)
##  ggplot2        * 3.3.6   2022-05-03 [1] CRAN (R 4.2.0)
##  glue             1.6.2   2022-02-24 [1] CRAN (R 4.2.0)
##  gtable           0.3.0   2019-03-25 [1] CRAN (R 4.2.0)
##  haven            2.5.0   2022-04-15 [1] CRAN (R 4.2.0)
##  here             1.0.1   2020-12-13 [1] CRAN (R 4.2.0)
##  highr            0.9     2021-04-16 [1] CRAN (R 4.2.0)
##  hms              1.1.1   2021-09-26 [1] CRAN (R 4.2.0)
##  htmltools        0.5.2   2021-08-25 [1] CRAN (R 4.2.0)
##  httr             1.4.3   2022-05-04 [1] CRAN (R 4.2.0)
##  jquerylib        0.1.4   2021-04-26 [1] CRAN (R 4.2.0)
##  jsonlite         1.8.0   2022-02-22 [1] CRAN (R 4.2.0)
##  knitr            1.39    2022-04-26 [1] CRAN (R 4.2.0)
##  labeling         0.4.2   2020-10-20 [1] CRAN (R 4.2.0)
##  lifecycle        1.0.1   2021-09-24 [1] CRAN (R 4.2.0)
##  lubridate        1.8.0   2021-10-07 [1] CRAN (R 4.2.0)
##  magrittr         2.0.3   2022-03-30 [1] CRAN (R 4.2.0)
##  memoise          2.0.1   2021-11-26 [1] CRAN (R 4.2.0)
##  modelr           0.1.8   2020-05-19 [1] CRAN (R 4.2.0)
##  munsell          0.5.0   2018-06-12 [1] CRAN (R 4.2.0)
##  pillar           1.7.0   2022-02-01 [1] CRAN (R 4.2.0)
##  pkgbuild         1.3.1   2021-12-20 [1] CRAN (R 4.2.0)
##  pkgconfig        2.0.3   2019-09-22 [1] CRAN (R 4.2.0)
##  pkgload          1.2.4   2021-11-30 [1] CRAN (R 4.2.0)
##  prettyunits      1.1.1   2020-01-24 [1] CRAN (R 4.2.0)
##  processx         3.5.3   2022-03-25 [1] CRAN (R 4.2.0)
##  ps               1.7.0   2022-04-23 [1] CRAN (R 4.2.0)
##  purrr          * 0.3.4   2020-04-17 [1] CRAN (R 4.2.0)
##  R6               2.5.1   2021-08-19 [1] CRAN (R 4.2.0)
##  rcfss          * 0.2.4   2022-06-08 [1] local
##  readr          * 2.1.2   2022-01-30 [1] CRAN (R 4.2.0)
##  readxl           1.4.0   2022-03-28 [1] CRAN (R 4.2.0)
##  remotes          2.4.2   2021-11-30 [1] CRAN (R 4.2.0)
##  reprex           2.0.1   2021-08-05 [1] CRAN (R 4.2.0)
##  rlang            1.0.2   2022-03-04 [1] CRAN (R 4.2.0)
##  rmarkdown        2.14    2022-04-25 [1] CRAN (R 4.2.0)
##  rprojroot        2.0.3   2022-04-02 [1] CRAN (R 4.2.0)
##  rstudioapi       0.13    2020-11-12 [1] CRAN (R 4.2.0)
##  rvest            1.0.2   2021-10-16 [1] CRAN (R 4.2.0)
##  sass             0.4.1   2022-03-23 [1] CRAN (R 4.2.0)
##  scales           1.2.0   2022-04-13 [1] CRAN (R 4.2.0)
##  sessioninfo      1.2.2   2021-12-06 [1] CRAN (R 4.2.0)
##  showtext         0.9-5   2022-02-09 [1] CRAN (R 4.2.0)
##  showtextdb       3.0     2020-06-04 [1] CRAN (R 4.2.0)
##  stringi          1.7.6   2021-11-29 [1] CRAN (R 4.2.0)
##  stringr        * 1.4.0   2019-02-10 [1] CRAN (R 4.2.0)
##  sysfonts         0.8.8   2022-03-13 [1] CRAN (R 4.2.0)
##  testthat         3.1.4   2022-04-26 [1] CRAN (R 4.2.0)
##  tibble         * 3.1.7   2022-05-03 [1] CRAN (R 4.2.0)
##  tidyr          * 1.2.0   2022-02-01 [1] CRAN (R 4.2.0)
##  tidyselect       1.1.2   2022-02-21 [1] CRAN (R 4.2.0)
##  tidyverse      * 1.3.1   2021-04-15 [1] CRAN (R 4.2.0)
##  tzdb             0.3.0   2022-03-28 [1] CRAN (R 4.2.0)
##  usethis          2.1.6   2022-05-25 [1] CRAN (R 4.2.0)
##  utf8             1.2.2   2021-07-24 [1] CRAN (R 4.2.0)
##  vctrs            0.4.1   2022-04-13 [1] CRAN (R 4.2.0)
##  withr            2.5.0   2022-03-03 [1] CRAN (R 4.2.0)
##  xaringanthemer   0.4.1   2021-11-21 [1] CRAN (R 4.2.0)
##  xfun             0.31    2022-05-10 [1] CRAN (R 4.2.0)
##  xml2             1.3.3   2021-11-30 [1] CRAN (R 4.2.0)
##  yaml             2.3.5   2022-02-21 [1] CRAN (R 4.2.0)
## 
##  [1] /Library/Frameworks/R.framework/Versions/4.2-arm64/Resources/library
## 
## ──────────────────────────────────────────────────────────────────────────────