Повернення значень з функцій

Вступ до написання функцій в R

Richie Cotton

Data Evangelist at DataCamp

Проста функція суми

simple_sum <- function(x) {

  if(anyNA(x)) {

    return(NA)

  }

  total <- 0
  for(value in x) {
    total <- total + value
  }
  total
}

simple_sum(c(0, 1, 3, 6, NA, 7))

NA

Знову про геометричне середнє

calc_geometric_mean <- function(x, na.rm = FALSE) {
  assert_is_numeric(x)
  if(any(is_non_positive(x), na.rm = TRUE)) {
    stop("x contains non-positive values, so the geometric mean makes no sense.")
  }
  na.rm <- coerce_to(use_first(na.rm), "logical")
  x %>%
    log() %>%
    mean(na.rm = na.rm) %>%
    exp()
}

Повертаємо NaN із попередженням

calc_geometric_mean <- function(x, na.rm = FALSE) {
  assert_is_numeric(x)
  if(any(is_non_positive(x), na.rm = TRUE)) {

    warning("x contains non-positive values, so the geometric mean makes no sense.")
    return(NaN)

  }
  na.rm <- coerce_to(use_first(na.rm), "logical")
  x %>%
    log() %>%
    mean(na.rm = na.rm) %>%
    exp()
}

Чому варто повертати раніше

Ви вже знаєте відповідь.
Вхідні дані — крайній випадок.

Приховування поверненого значення

simple_sum <- function(x) {
  if(anyNA(x)) {
    return(NA)
  }
  total <- 0
  for(value in x) {
    total <- total + value
  }
  total
}

simple_sum(c(0, 1, 3, 6, 2, 7))

Приховування поверненого значення

simple_sum <- function(x) {
  if(anyNA(x)) {
    return(NA)
  }
  total <- 0
  for(value in x) {
    total <- total + value
  }

  invisible(total)

}

simple_sum(c(0, 1, 3, 6, 2, 7))

Багато графіків непомітно щось повертають

ggplot(snake_river_visits, aes(n_visits)) +
  geom_histogram(binwidth = 10)

Гістограма відвідувань Snake River. Розподіл правостиснений: більшість у лівій смузі — менше 5 відвідувань на рік. Найбільше значення — понад 300 відвідувань на рік.

Багато графіків непомітно щось повертають

srv_hist <- ggplot(snake_river_visits, aes(n_visits)) +
  geom_histogram(binwidth = 10)

str(srv_hist, max.level = 0)

List of 9
 - attr(*, "class")= chr [1:2] "gg" "ggplot"

Ту саму гістограму відвідувань Snake River з попереднього слайда показано знову.

Давайте потренуємось!

Вступ до написання функцій в R