lapply 和 sapply

本课属于 Coddy R 学习旅程的“逻辑与流程”单元。第 32 课，共 64 课。

lapply(x, f) 会对 x 的每个元素调用 f，并将结果作为列表返回，每个输入对应一个元素：

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

输出：

[1] 3
[1] 9

sapply() 的作用相同，但会简化结果：当每次调用都返回一个值时，你会得到一个普通向量，而不是列表：

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

输出：

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

对于字符向量，sapply() 会将结果命名为对应的输入项。添加 USE.NAMES = FALSE 可得到普通向量：

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

输出：

   tea coffee 
     3      6 
[1] 3 6

这些函数可以替代逐个元素构建结果的循环。下面两个版本都会计算每个单词中的元音字母数量：

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

输出：

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

挑战

简单

完成 vowel_counts(words)。每个单词返回一行，banana: 3，其中包含该单词中的元音字母数量（a、e、i、o、u）。使用 sapply() 统计所有单词中的元音字母。

提供的代码会读取一行以逗号分隔的小写单词，并打印返回的每一行。

亲手试试

vowel_counts <- function(words) {
  # 在这里编写你的代码
  character(0)
}

# 提供的输入输出代码：保持原样
input <- suppressWarnings(readLines(file("stdin")))
cat(vowel_counts(strsplit(input[1], ",")[[1]]), sep = "\n")