R为列表的每个元素提供不同的功能

时间:2015-07-06 18:38:37

标签: r function lapply

我有一个列表(在R中),其中元素是不同的数据类型,例如,第一个元素是数字,第二个元素是字符。我想对每个元素应用不同的函数。例如,在下面的代码中,我尝试仅将sum函数应用于第一个元素,将length函数仅应用于第二个元素。有没有办法将不同的函数应用于列表的每个元素(不分解列表)?

data <- list(
  A = rnorm(10),
  B = letters[1:10]
)

lapply(data, list(sum, length))

mapply(function(x) sum, length, data)

2 个答案:

答案 0 :(得分:16)

怎么样

mapply(function(a,b) b(a), data, list(sum, length))

请注意,我们也将mapply中的函数放在列表中。

答案 1 :(得分:6)

我会做像

这样的事情
sapply( data, function(x) (if(is.character(x)) length else sum)(x) )

复杂的替代方案。如果速度是一个问题,vapply应该更快:

vapply( data, function(x) (if(is.character(x)) length else sum)(x), numeric(1) )

如果您需要多次使用length,则可以快速使用lengths(在R 3.2.0 +中提供):

res          <- lengths(data)
get_sum      <- !sapply(data,is.character)
res[get_sum] <- sapply(data[get_sum],sum)