lapply i sapply

Część sekcji Logika i przepływ sterowania ścieżki R w Coddy. Lekcja 32 z 64.

lapply(x, f) wywołuje f dla każdego elementu x i zwraca wyniki w postaci listy, po jednym elemencie na każde wejście:

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

Wynik:

[1] 3
[1] 9

sapply() robi to samo i upraszcza wynik: gdy każde wywołanie zwraca jedną wartość, otrzymujesz zwykły wektor zamiast listy:

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

Wynik:

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

W przypadku wektora znakowego sapply() nadaje wynikom nazwy odpowiadające wartościom wejściowym. Dodaj USE.NAMES = FALSE, aby uzyskać zwykły wektor:

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

Wynik:

   tea coffee 
     3      6 
[1] 3 6

Te funkcje zastępują pętlę, która buduje wynik, dodając po jednym elemencie naraz. Obie poniższe wersje obliczają, ile samogłosek ma każde słowo:

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))

Wynik:

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

Wyzwanie

Łatwy

Uzupełnij vowel_counts(words). Zwróć po jednym wierszu dla każdego słowa, banana: 3, z liczbą samogłosek (a, e, i, o, u) w słowie. Użyj sapply(), aby policzyć samogłoski we wszystkich słowach.

Dostarczony kod odczytuje słowa jako wiersz zawierający słowa zapisane małymi literami i rozdzielone przecinkami, a następnie wyświetla każdy zwrócony wiersz.

Spróbuj swoich sił

vowel_counts <- function(words) {
  # Wpisz tutaj swój kod
  character(0)
}

# Dostarczony kod wejścia/wyjścia: pozostaw go bez zmian
input <- suppressWarnings(readLines(file("stdin")))
cat(vowel_counts(strsplit(input[1], ",")[[1]]), sep = "\n")