ifelse() Vetorizado

Parte da seção Lógica e Fluxo do Journey de R da Coddy. Lição 48 de 64.

if testa uma condição. Dado um vetor com vários valores, ele para com o erro the condition has length > 1. ifelse(test, yes, no) funciona elemento por elemento e retorna um resultado por elemento:

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

Saída:

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

yes e no podem ser vetores do mesmo comprimento, portanto cada elemento pode obter seu próprio valor calculado:

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

Saída:

[1]  40 108  75

Aninhe chamadas de ifelse() para obter mais de dois resultados. As condições são verificadas em ordem para cada elemento:

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

Saída:

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

ifelse() é útil para uma nova coluna de um data frame, em que cada linha precisa de seu próprio rótulo:

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

Saída:

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

Desafio

Fácil

Complete shipping(weights). Retorne o custo de envio de cada pacote, unidos por , : pacotes de até 1 kg custam 4, de até 5 kg custam 9, e pacotes mais pesados custam 2 por kg. Use ifelse(), não um loop.

O código fornecido lê os pesos como uma linha separada por vírgulas e imprime o valor retornado.

Experimente você mesmo

shipping <- function(weights) {
  # Escreva seu código aqui
  ""
}

# Código de entrada/saída fornecido: mantenha-o como está
input <- suppressWarnings(readLines(file("stdin")))
cat(shipping(as.numeric(strsplit(input[1], ",")[[1]])), sep = "\n")