class: center, middle, inverse, title-slide .title[ # Hands on Data Wrangling
🔧 ] .author[ ### S. Mason Garrison ] --- layout: true <div class="my-footer"> <span> <a href="https://DataScience4Psych.github.io/DataScience4Psych/" target="_blank">Data Science for Psychologists</a> </span> </div> --- class: middle # Working with a single data frame --- class: middle .hand[You...] .huge-blue[have] .hand[a single data frame] .huge-pink[want] .hand[to slice it, and dice it, and juice it, and process it] --- ## `select` to keep variables .medi[ ``` r hotels %>% * select(hotel, lead_time) ``` ``` ## # A tibble: 119,390 × 2 ## hotel lead_time ## <chr> <dbl> ## 1 Resort Hotel 342 ## 2 Resort Hotel 737 ## 3 Resort Hotel 7 ## 4 Resort Hotel 13 ## 5 Resort Hotel 14 ## 6 Resort Hotel 14 ## 7 Resort Hotel 0 ## 8 Resort Hotel 9 ## 9 Resort Hotel 85 ## 10 Resort Hotel 75 ## # ℹ 119,380 more rows ``` ] --- ## `select` to exclude variables .small[ ``` r hotels %>% * select(-agent) ``` ``` ## # A tibble: 119,390 × 31 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 Resort Hotel 0 342 2015 ## 2 Resort Hotel 0 737 2015 ## 3 Resort Hotel 0 7 2015 ## 4 Resort Hotel 0 13 2015 ## 5 Resort Hotel 0 14 2015 ## 6 Resort Hotel 0 14 2015 ## 7 Resort Hotel 0 0 2015 ## 8 Resort Hotel 0 9 2015 ## 9 Resort Hotel 1 85 2015 ## 10 Resort Hotel 1 75 2015 ## # ℹ 119,380 more rows ## # ℹ 27 more variables: arrival_date_month <chr>, ## # arrival_date_week_number <dbl>, ## # arrival_date_day_of_month <dbl>, ## # stays_in_weekend_nights <dbl>, stays_in_week_nights <dbl>, ## # adults <dbl>, children <dbl>, babies <dbl>, meal <chr>, ## # country <chr>, market_segment <chr>, … ``` ] --- ## `select` a range of variables .medi[ ``` r hotels %>% * select(hotel:arrival_date_month) ``` ``` ## # A tibble: 119,390 × 5 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 Resort Hotel 0 342 2015 ## 2 Resort Hotel 0 737 2015 ## 3 Resort Hotel 0 7 2015 ## 4 Resort Hotel 0 13 2015 ## 5 Resort Hotel 0 14 2015 ## 6 Resort Hotel 0 14 2015 ## 7 Resort Hotel 0 0 2015 ## 8 Resort Hotel 0 9 2015 ## 9 Resort Hotel 1 85 2015 ## 10 Resort Hotel 1 75 2015 ## # ℹ 119,380 more rows ## # ℹ 1 more variable: arrival_date_month <chr> ``` ] --- ## `select` variables with certain characteristics .medi[ ``` r hotels %>% * select(starts_with("arrival")) ``` ``` ## # A tibble: 119,390 × 4 ## arrival_date_year arrival_date_month arrival_date_week_number ## <dbl> <chr> <dbl> ## 1 2015 July 27 ## 2 2015 July 27 ## 3 2015 July 27 ## 4 2015 July 27 ## 5 2015 July 27 ## 6 2015 July 27 ## 7 2015 July 27 ## 8 2015 July 27 ## 9 2015 July 27 ## 10 2015 July 27 ## # ℹ 119,380 more rows ## # ℹ 1 more variable: arrival_date_day_of_month <dbl> ``` ] --- ## `select` variables with certain characteristics .medi[ ``` r hotels %>% * select(ends_with("type")) ``` ``` ## # A tibble: 119,390 × 4 ## reserved_room_type assigned_room_type deposit_type ## <chr> <chr> <chr> ## 1 C C No Deposit ## 2 C C No Deposit ## 3 A C No Deposit ## 4 A A No Deposit ## 5 A A No Deposit ## 6 A A No Deposit ## 7 C C No Deposit ## 8 C C No Deposit ## 9 A A No Deposit ## 10 D D No Deposit ## # ℹ 119,380 more rows ## # ℹ 1 more variable: customer_type <chr> ``` ] --- ## Select helpers - `starts_with()`: Starts with a prefix - `ends_with()`: Ends with a suffix - `contains()`: Contains a literal string - `num_range()`: Matches a numerical range like x01, x02, x03 - `one_of()`: Matches variable names in a character vector - `everything()`: Matches all variables - `last_col()`: Select last variable, possibly with an offset - `matches()`: Matches a regular expression (a sequence of symbols/characters expressing a string/pattern to be searched for within text) .footnote[ See help for any of these functions for more info, e.g. `?everything`. ] --- ## `arrange` in ascending / descending order .pull-left[ ``` r hotels %>% select(adults, children, babies) %>% * arrange(babies) ``` ``` ## # A tibble: 119,390 × 3 ## adults children babies ## <dbl> <dbl> <dbl> ## 1 2 0 0 ## 2 2 0 0 ## 3 1 0 0 ## 4 1 0 0 ## 5 2 0 0 ## 6 2 0 0 ## 7 2 0 0 ## 8 2 0 0 ## 9 2 0 0 ## 10 2 0 0 ## # ℹ 119,380 more rows ``` ] .pull-right[ ``` r hotels %>% select(adults, children, babies) %>% * arrange(desc(babies)) ``` ``` ## # A tibble: 119,390 × 3 ## adults children babies ## <dbl> <dbl> <dbl> ## 1 2 0 10 ## 2 1 0 9 ## 3 2 0 2 ## 4 2 0 2 ## 5 2 0 2 ## 6 2 0 2 ## 7 2 0 2 ## 8 2 0 2 ## 9 2 0 2 ## 10 2 0 2 ## # ℹ 119,380 more rows ``` ] --- ## `slice` for certain row numbers .midi[ ``` r # first five hotels %>% * slice(1:5) ``` ``` ## # A tibble: 5 × 32 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 Resort Hotel 0 342 2015 ## 2 Resort Hotel 0 737 2015 ## 3 Resort Hotel 0 7 2015 ## 4 Resort Hotel 0 13 2015 ## 5 Resort Hotel 0 14 2015 ## # ℹ 28 more variables: arrival_date_month <chr>, ## # arrival_date_week_number <dbl>, ## # arrival_date_day_of_month <dbl>, ## # stays_in_weekend_nights <dbl>, stays_in_week_nights <dbl>, ## # adults <dbl>, children <dbl>, babies <dbl>, meal <chr>, ## # country <chr>, market_segment <chr>, ## # distribution_channel <chr>, is_repeated_guest <dbl>, … ``` ] --- .tip[ + In R, you can use the `#` (hashtag or pound sign, depending on your age 😜) for adding comments to your code. Any text following `#` will be printed as is, and won't be run as R code. + This symbol is useful for leaving comments in your code and for temporarily disabling certain lines of code while debugging. ] -- .xsmall[ ``` r hotels %>% # slice the first five rows # this line is a comment #select(hotel) %>% # this one doesn't run slice(1:5) # this line runs ``` ``` ## # A tibble: 5 × 32 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 Resort Hotel 0 342 2015 ## 2 Resort Hotel 0 737 2015 ## 3 Resort Hotel 0 7 2015 ## 4 Resort Hotel 0 13 2015 ## 5 Resort Hotel 0 14 2015 ## # ℹ 28 more variables: arrival_date_month <chr>, ## # arrival_date_week_number <dbl>, ## # arrival_date_day_of_month <dbl>, ## # stays_in_weekend_nights <dbl>, stays_in_week_nights <dbl>, ## # adults <dbl>, children <dbl>, babies <dbl>, meal <chr>, ## # country <chr>, market_segment <chr>, ## # distribution_channel <chr>, is_repeated_guest <dbl>, … ``` ] --- ## `slice` for certain row numbers .small[ ``` r # last five last_row <- nrow(hotels) # nrow() gives the number of rows in a data frame hotels %>% * slice((last_row - 4):last_row) ``` ``` ## # A tibble: 5 × 32 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 City Hotel 0 23 2017 ## 2 City Hotel 0 102 2017 ## 3 City Hotel 0 34 2017 ## 4 City Hotel 0 109 2017 ## 5 City Hotel 0 205 2017 ## # ℹ 28 more variables: arrival_date_month <chr>, ## # arrival_date_week_number <dbl>, ## # arrival_date_day_of_month <dbl>, ## # stays_in_weekend_nights <dbl>, stays_in_week_nights <dbl>, ## # adults <dbl>, children <dbl>, babies <dbl>, meal <chr>, ## # country <chr>, market_segment <chr>, ## # distribution_channel <chr>, is_repeated_guest <dbl>, … ``` ] --- ## `filter` to select a subset of rows .small[ ``` r # bookings in City Hotels hotels %>% * filter(hotel == "City Hotel") ``` ``` ## # A tibble: 79,330 × 32 ## hotel is_canceled lead_time arrival_date_year ## <chr> <dbl> <dbl> <dbl> ## 1 City Hotel 0 6 2015 ## 2 City Hotel 1 88 2015 ## 3 City Hotel 1 65 2015 ## 4 City Hotel 1 92 2015 ## 5 City Hotel 1 100 2015 ## 6 City Hotel 1 79 2015 ## 7 City Hotel 0 3 2015 ## 8 City Hotel 1 63 2015 ## 9 City Hotel 1 62 2015 ## 10 City Hotel 1 62 2015 ## # ℹ 79,320 more rows ## # ℹ 28 more variables: arrival_date_month <chr>, ## # arrival_date_week_number <dbl>, ## # arrival_date_day_of_month <dbl>, ## # stays_in_weekend_nights <dbl>, stays_in_week_nights <dbl>, ## # adults <dbl>, children <dbl>, babies <dbl>, meal <chr>, ## # country <chr>, market_segment <chr>, … ``` ] --- ## `filter` for many conditions at once .medi[ ``` r hotels %>% filter( * adults == 0, * children >= 1 ) %>% select(adults, babies, children) ``` ``` ## # A tibble: 223 × 3 ## adults babies children ## <dbl> <dbl> <dbl> ## 1 0 0 3 ## 2 0 0 2 ## 3 0 0 2 ## 4 0 0 2 ## 5 0 0 2 ## 6 0 0 3 ## 7 0 1 2 ## 8 0 0 2 ## 9 0 0 2 ## 10 0 0 2 ## # ℹ 213 more rows ``` ] --- ## `filter` for more complex conditions .midi[ ``` r # bookings with no adults and some children or babies in the room hotels %>% filter( adults == 0, * children >= 1 | babies >= 1 # | means or ) %>% select(adults, babies, children) ``` ``` ## # A tibble: 223 × 3 ## adults babies children ## <dbl> <dbl> <dbl> ## 1 0 0 3 ## 2 0 0 2 ## 3 0 0 2 ## 4 0 0 2 ## 5 0 0 2 ## 6 0 0 3 ## 7 0 1 2 ## 8 0 0 2 ## 9 0 0 2 ## 10 0 0 2 ## # ℹ 213 more rows ``` ] --- ## Logical operators in R <br> operator | definition || operator | definition ------------|------------------------------||--------------|---------------- `<` | less than ||`x` | `y` | `x` OR `y` `<=` | less than or equal to ||`is.na(x)` | test if `x` is `NA` `>` | greater than ||`!is.na(x)` | test if `x` is not `NA` `>=` | greater than or equal to ||`x %in% y` | test if `x` is in `y` `==` | exactly equal to ||`!(x %in% y)` | test if `x` is not in `y` `!=` | not equal to ||`!x` | not `x` `x & y` | `x` AND `y` || | --- .your-turn[ Time to actually play around with the Hotels dataset! - Go to class git repo ([github.com/DataScience4Psych](https://github.com/DataScience4Psych)) and start `AE 04 - Hotels + Data wrangling`. - Open the R Markdown document and complete Exercises 1 - 4. ] --- class: middle # Pause ... --- class: middle # Moving Along... --- ## `distinct` to filter for unique rows ... and `arrange` to order alphabetically .small[ .pull-left[ ``` r hotels %>% * distinct(market_segment) %>% arrange(market_segment) ``` ``` ## # A tibble: 8 × 1 ## market_segment ## <chr> ## 1 Aviation ## 2 Complementary ## 3 Corporate ## 4 Direct ## 5 Groups ## 6 Offline TA/TO ## 7 Online TA ## 8 Undefined ``` ] .pull-right[ ``` r hotels %>% * distinct(hotel, market_segment) %>% arrange(hotel, market_segment) ``` ``` ## # A tibble: 14 × 2 ## hotel market_segment ## <chr> <chr> ## 1 City Hotel Aviation ## 2 City Hotel Complementary ## 3 City Hotel Corporate ## 4 City Hotel Direct ## 5 City Hotel Groups ## 6 City Hotel Offline TA/TO ## 7 City Hotel Online TA ## 8 City Hotel Undefined ## 9 Resort Hotel Complementary ## 10 Resort Hotel Corporate ## 11 Resort Hotel Direct ## 12 Resort Hotel Groups ## 13 Resort Hotel Offline TA/TO ## 14 Resort Hotel Online TA ``` ] ] --- ## `count` to create frequency tables .pull-left[ ``` r # alphabetical order by default hotels %>% * count(market_segment) ``` ``` ## # A tibble: 8 × 2 ## market_segment n ## <chr> <int> ## 1 Aviation 237 ## 2 Complementary 743 ## 3 Corporate 5295 ## 4 Direct 12606 ## 5 Groups 19811 ## 6 Offline TA/TO 24219 ## 7 Online TA 56477 ## 8 Undefined 2 ``` ] -- .pull-right[ ``` r # descending frequency order hotels %>% * count(market_segment, sort = TRUE) ``` ``` ## # A tibble: 8 × 2 ## market_segment n ## <chr> <int> ## 1 Online TA 56477 ## 2 Offline TA/TO 24219 ## 3 Groups 19811 ## 4 Direct 12606 ## 5 Corporate 5295 ## 6 Complementary 743 ## 7 Aviation 237 ## 8 Undefined 2 ``` ] --- ## `count` and `arrange` .pull-left[ ``` r # ascending frequency order hotels %>% count(market_segment) %>% * arrange(n) ``` ``` ## # A tibble: 8 × 2 ## market_segment n ## <chr> <int> ## 1 Undefined 2 ## 2 Aviation 237 ## 3 Complementary 743 ## 4 Corporate 5295 ## 5 Direct 12606 ## 6 Groups 19811 ## 7 Offline TA/TO 24219 ## 8 Online TA 56477 ``` ] .pull-right[ ``` r # descending frequency order # just like adding sort = TRUE hotels %>% count(market_segment) %>% * arrange(desc(n)) ``` ``` ## # A tibble: 8 × 2 ## market_segment n ## <chr> <int> ## 1 Online TA 56477 ## 2 Offline TA/TO 24219 ## 3 Groups 19811 ## 4 Direct 12606 ## 5 Corporate 5295 ## 6 Complementary 743 ## 7 Aviation 237 ## 8 Undefined 2 ``` ] --- ## `count` for multiple variables .medi[ ``` r hotels %>% * count(hotel, market_segment) ``` ``` ## # A tibble: 14 × 3 ## hotel market_segment n ## <chr> <chr> <int> ## 1 City Hotel Aviation 237 ## 2 City Hotel Complementary 542 ## 3 City Hotel Corporate 2986 ## 4 City Hotel Direct 6093 ## 5 City Hotel Groups 13975 ## 6 City Hotel Offline TA/TO 16747 ## 7 City Hotel Online TA 38748 ## 8 City Hotel Undefined 2 ## 9 Resort Hotel Complementary 201 ## 10 Resort Hotel Corporate 2309 ## 11 Resort Hotel Direct 6513 ## 12 Resort Hotel Groups 5836 ## 13 Resort Hotel Offline TA/TO 7472 ## 14 Resort Hotel Online TA 17729 ``` ] --- ## order matters when you `count` .midi[ .pull-left[ ``` r # hotel type first hotels %>% * count(hotel, market_segment) ``` ``` ## # A tibble: 14 × 3 ## hotel market_segment n ## <chr> <chr> <int> ## 1 City Hotel Aviation 237 ## 2 City Hotel Complementary 542 ## 3 City Hotel Corporate 2986 ## 4 City Hotel Direct 6093 ## 5 City Hotel Groups 13975 ## 6 City Hotel Offline TA/TO 16747 ## 7 City Hotel Online TA 38748 ## 8 City Hotel Undefined 2 ## 9 Resort Hotel Complementary 201 ## 10 Resort Hotel Corporate 2309 ## 11 Resort Hotel Direct 6513 ## 12 Resort Hotel Groups 5836 ## 13 Resort Hotel Offline TA/TO 7472 ## 14 Resort Hotel Online TA 17729 ``` ] .pull-right[ ``` r # market segment first hotels %>% * count(market_segment, hotel) ``` ``` ## # A tibble: 14 × 3 ## market_segment hotel n ## <chr> <chr> <int> ## 1 Aviation City Hotel 237 ## 2 Complementary City Hotel 542 ## 3 Complementary Resort Hotel 201 ## 4 Corporate City Hotel 2986 ## 5 Corporate Resort Hotel 2309 ## 6 Direct City Hotel 6093 ## 7 Direct Resort Hotel 6513 ## 8 Groups City Hotel 13975 ## 9 Groups Resort Hotel 5836 ## 10 Offline TA/TO City Hotel 16747 ## 11 Offline TA/TO Resort Hotel 7472 ## 12 Online TA City Hotel 38748 ## 13 Online TA Resort Hotel 17729 ## 14 Undefined City Hotel 2 ``` ] ] --- .your-turn[ Time to actually play around with the Hotels dataset! - Go to class git repo ([github.com/DataScience4Psych](https://github.com/DataScience4Psych)) and start `AE 04 - Hotels + Data wrangling`. - Open the R Markdown document and complete Exercises 5 and 6. ] --- class: middle # Pause until you're ready ... --- ## `mutate` to add a new variable .medi[ ``` r hotels %>% * mutate(little_ones = children + babies) %>% select(children, babies, little_ones) %>% arrange(desc(little_ones)) ``` ``` ## # A tibble: 119,390 × 3 ## children babies little_ones ## <dbl> <dbl> <dbl> ## 1 10 0 10 ## 2 0 10 10 ## 3 0 9 9 ## 4 2 1 3 ## 5 2 1 3 ## 6 2 1 3 ## 7 3 0 3 ## 8 2 1 3 ## 9 2 1 3 ## 10 3 0 3 ## # ℹ 119,380 more rows ``` ] --- ## Little ones in resort and city hotels .midi[ .pull-left[ ``` r # Resort Hotel hotels %>% mutate(little_ones = children + babies) %>% filter( little_ones >= 1, hotel == "Resort Hotel" ) %>% select(hotel, little_ones) ``` ``` ## # A tibble: 3,929 × 2 ## hotel little_ones ## <chr> <dbl> ## 1 Resort Hotel 1 ## 2 Resort Hotel 2 ## 3 Resort Hotel 2 ## 4 Resort Hotel 2 ## 5 Resort Hotel 1 ## 6 Resort Hotel 1 ## 7 Resort Hotel 2 ## 8 Resort Hotel 2 ## 9 Resort Hotel 1 ## 10 Resort Hotel 1 ## # ℹ 3,919 more rows ``` ] .pull-right[ ``` r # City Hotel hotels %>% mutate(little_ones = children + babies) %>% filter( little_ones > 1, hotel == "City Hotel" ) %>% select(hotel, little_ones) ``` ``` ## # A tibble: 2,140 × 2 ## hotel little_ones ## <chr> <dbl> ## 1 City Hotel 2 ## 2 City Hotel 2 ## 3 City Hotel 2 ## 4 City Hotel 2 ## 5 City Hotel 2 ## 6 City Hotel 2 ## 7 City Hotel 2 ## 8 City Hotel 2 ## 9 City Hotel 2 ## 10 City Hotel 3 ## # ℹ 2,130 more rows ``` ] ] --- .question[ What is happening in the following chunk? ] .midi[ ``` r hotels %>% mutate(little_ones = children + babies) %>% count(hotel, little_ones) %>% mutate(prop = n / sum(n)) ``` ``` ## # A tibble: 12 × 4 ## hotel little_ones n prop ## <chr> <dbl> <int> <dbl> ## 1 City Hotel 0 73923 0.619 ## 2 City Hotel 1 3263 0.0273 ## 3 City Hotel 2 2056 0.0172 ## 4 City Hotel 3 82 0.000687 ## 5 City Hotel 9 1 0.00000838 ## 6 City Hotel 10 1 0.00000838 ## 7 City Hotel NA 4 0.0000335 ## 8 Resort Hotel 0 36131 0.303 ## 9 Resort Hotel 1 2183 0.0183 ## 10 Resort Hotel 2 1716 0.0144 ## 11 Resort Hotel 3 29 0.000243 ## 12 Resort Hotel 10 1 0.00000838 ``` ] --- # `summarize` for summary stats ``` r # mean average daily rate for all bookings hotels %>% * summarize(mean_adr = mean(adr)) ``` ``` ## # A tibble: 1 × 1 ## mean_adr ## <dbl> ## 1 102. ``` -- <br> .tip[ `summarize()` changes the data frame entirely, it collapses rows down to a single summary statistic, and removes all columns that are irrelevant to the calculation. ] --- .tip[ `summarize()` also lets you get away with being sloppy and not naming your new column, but that's not recommended! ] .midi[ ❌ ``` r hotels %>% summarize(mean(adr)) ``` ``` ## # A tibble: 1 × 1 ## `mean(adr)` ## <dbl> ## 1 102. ``` ✅ ``` r hotels %>% summarize(mean_adr = mean(adr)) ``` ``` ## # A tibble: 1 × 1 ## mean_adr ## <dbl> ## 1 102. ``` ] --- # `group_by` for grouped operations ``` r # mean average daily rate for all booking at city and resort hotels hotels %>% * group_by(hotel) %>% summarize(mean_adr = mean(adr)) ``` ``` ## # A tibble: 2 × 2 ## hotel mean_adr ## <chr> <dbl> ## 1 City Hotel 105. ## 2 Resort Hotel 95.0 ``` --- ## Calculating frequencies The following two give the same result, so `count` is simply short for `group_by` then determine frequencies .pull-left[ ``` r hotels %>% group_by(hotel) %>% summarize(n = n()) ``` ``` ## # A tibble: 2 × 2 ## hotel n ## <chr> <int> ## 1 City Hotel 79330 ## 2 Resort Hotel 40060 ``` ] .pull-right[ ``` r hotels %>% count(hotel) ``` ``` ## # A tibble: 2 × 2 ## hotel n ## <chr> <int> ## 1 City Hotel 79330 ## 2 Resort Hotel 40060 ``` ] --- # Multiple summary statistics `summarize` can be used for multiple summary statistics as well ``` r hotels %>% summarize( min_adr = min(adr), mean_adr = mean(adr), median_adr = median(adr), max_adr = max(adr) ) ``` ``` ## # A tibble: 1 × 4 ## min_adr mean_adr median_adr max_adr ## <dbl> <dbl> <dbl> <dbl> ## 1 -6.38 102. 94.6 5400 ``` --- .your-turn[ Time to actually play around with the Hotels dataset! - Go to class git repo ([github.com/DataScience4Psych](https://github.com/DataScience4Psych)) and start `AE 04 - Hotels + Data wrangling`. - Open the R Markdown document and complete Exercises 7 and 8. ] --- # Sources - Mine Çetinkaya-Rundel's Data Science in a Box ([link](https://datasciencebox.org/)) --- class: middle # Wrapping Up...