我有一个列表(在R中),其中元素是不同的数据类型,例如,第一个元素是数字,第二个元素是字符。我想对每个元素应用不同的函数。例如,在下面的代码中,我尝试仅将sum函数应用于第一个元素,将length函数仅应用于第二个元素。有没有办法将不同的函数应用于列表的每个元素(不分解列表)?
data <- list(
A = rnorm(10),
B = letters[1:10]
)
lapply(data, list(sum, length))
mapply(function(x) sum, length, data)
答案 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)