lapply e sapply

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

lapply(x, f) chama f para cada elemento de x e retorna os resultados em uma lista, um elemento por entrada:

squares <- lapply(1:3, function(i) i^2)
print(length(squares))
print(squares[[3]])

Saída:

[1] 3
[1] 9

sapply() faz a mesma coisa e simplifica o resultado: quando cada chamada retorna um valor, você obtém um vetor simples em vez de uma lista:

print(sapply(1:3, function(i) i^2))
print(sapply(c(4, 9, 16), sqrt))

Saída:

[1] 1 4 9
[1] 2 3 4

Em um vetor de caracteres, sapply() nomeia os resultados de acordo com as entradas. Adicione USE.NAMES = FALSE para obter um vetor simples:

words <- c("tea", "coffee")
print(sapply(words, nchar))
print(sapply(words, nchar, USE.NAMES = FALSE))

Saída:

   tea coffee 
     3      6 
[1] 3 6

Essas funções substituem o loop que constrói um resultado um elemento por vez. Ambas as versões abaixo calculam quantas vogais cada palavra tem:

words <- c("banana", "sky", "idea")
vowels <- function(w) sum(strsplit(w, "")[[1]] %in% c("a", "e", "i", "o", "u"))
counts <- c()
for (w in words) counts <- c(counts, vowels(w))
print(counts)
print(sapply(words, vowels, USE.NAMES = FALSE))

Saída:

[1] 3 0 3
[1] 3 0 3

Desafio

Fácil

Complete vowel_counts(words). Retorne uma linha por palavra, banana: 3, com o número de vogais (a, e, i, o, u) na palavra. Use sapply() para contar as vogais de todas as palavras.

O código fornecido lê as palavras como uma linha separada por vírgulas de palavras em minúsculas e imprime cada linha retornada.

Experimente você mesmo

vowel_counts <- function(words) {
  # Escreva seu código aqui
  character(0)
}

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