Menu
Coddy logo textTech

Векторизованный ifelse()

Часть раздела Логика и потоки выполнения путешествия по R на Coddy. Урок 48 из 64.

if проверяет одно условие. Если ему передать вектор из нескольких значений, он завершится с ошибкой the condition has length > 1. ifelse(test, yes, no) работает поэлементно и возвращает по одному результату для каждого элемента:

temps <- c(12, 25, 31, 18)
print(ifelse(temps > 20, "warm", "cool"))

Вывод:

[1] "cool" "warm" "warm" "cool"

yes и no могут быть векторами одинаковой длины, поэтому каждый элемент может получить собственное вычисленное значение:

price <- c(40, 120, 75)
final <- ifelse(price > 100, price * 0.9, price)
print(final)

Результат:

[1]  40 108  75

Вкладывайте вызовы ifelse() для получения более двух результатов. Условия проверяются по порядку для каждого элемента:

scores <- c(95, 72, 48, 81)
grade <- ifelse(scores >= 90, "A", ifelse(scores >= 70, "B", "C"))
print(grade)

Результат:

[1] "A" "B" "C" "B"

ifelse() полезна для нового столбца фрейма данных, где каждой строке нужна собственная метка:

df <- data.frame(item = c("tea", "oil"), stock = c(0, 8))
df$status <- ifelse(df$stock == 0, "sold out", "available")
print(df)

Результат:

  item stock    status
1  tea     0  sold out
2  oil     8 available
challenge icon

Задание

Легко

Дополните shipping(weights). Верните стоимость доставки каждой посылки, объединённую с помощью , : посылки весом до 1 кг стоят 4, до 5 кг — 9, а более тяжёлые посылки стоят 2 за каждый кг. Используйте ifelse(), а не цикл.

Предоставленный код считывает веса в виде строки, разделённой запятыми, и выводит возвращённое значение.

Попробуйте сами

shipping <- function(weights) {
  # Напишите ваш код здесь
  ""
}

# Предоставленный код ввода/вывода: оставьте его как есть
input <- suppressWarnings(readLines(file("stdin")))
cat(shipping(as.numeric(strsplit(input[1], ",")[[1]])), sep = "\n")
quiz iconПроверьте себя

В этом уроке есть небольшой тест. Начните урок, чтобы ответить на вопросы и сохранить прогресс.

Все уроки раздела Логика и потоки выполнения

Потренируйтесь самостоятельно: Онлайн-компилятор R