在ifelse中使用Reduce / do.call

时间:2012-08-03 15:00:32

标签: r higher-order-functions

这纯粹是一种好奇心(更多地了解Reduce)。有更好的方法来实现我正在做的事情,我对它们不感兴趣。

有些人使用一系列嵌套ifelse命令来重新编码/查找某些内容。也许它看起来像这样:

set.seed(10); x <- sample(letters[1:10], 300, T)
ifelse(x=="a", 1, 
    ifelse(x=="b", 2,
    ifelse(x=="c", 3, 
    ifelse(x=="d", 4, 5))))

有没有办法可以do.callReduce使用ifelse来更有效地完成工作?

2 个答案:

答案 0 :(得分:3)

试试这个:

> library(gsubfn)
> strapply(x, ".", list(a = 1, b = 2, c = 3, d = 4, 5), simplify = TRUE)
  [1] 5 4 5 5 1 3 3 3 5 5 5 5 2 5 4 5 1 3 4 5 5 5 5 4 5 5 5 3 5 4 5 1 2 5 5 5 5
 [38] 5 5 5 3 3 1 5 3 2 1 5 2 5 4 5 3 5 2 5 5 5 4 5 1 2 5 4 5 5 5 5 1 3 1 5 5 5
 [75] 1 5 4 5 3 3 5 5 3 5 3 1 5 3 2 2 5 5 5 5 4 5 3 5 5 1 4 1 4 5 5 5 5 5 5 5 5
[112] 5 2 5 5 5 3 5 5 5 2 4 4 5 3 3 5 4 5 5 5 1 5 3 4 3 5 5 2 5 5 3 1 5 2 5 5 5
[149] 1 5 5 2 1 2 4 2 2 3 5 2 5 5 5 5 5 3 5 5 5 5 5 5 5 5 5 5 2 3 5 4 4 2 5 5 5
[186] 5 5 5 5 2 1 1 1 5 5 5 5 3 5 5 3 5 5 5 2 5 5 5 3 5 5 5 5 5 1 5 5 5 5 2 2 5
[223] 5 5 4 3 4 5 5 4 5 5 5 3 5 3 5 5 5 5 4 5 5 1 5 5 2 5 5 5 2 5 5 3 2 5 4 5 2
[260] 5 5 3 5 5 1 4 3 5 4 5 2 5 5 3 5 5 5 5 5 1 1 5 2 5 1 5 5 5 5 5 5 5 5 5 5 5
[297] 5 1 5 2

答案 1 :(得分:0)

这是一次尝试。它既不漂亮也不使用ifelse:

f <- function(w,s) {
  if(is.null(s$old)) 
    w$output[is.na(w$output)] <- s$new
  else 
    w$output[w$input==s$old] <- s$new
  return(w)
}

set.seed(10); x <- sample(letters[1:10], 300, T)

subst <- list(
  list(old="a", new=1), 
  list(old="b", new=2),
  list(old="c", new=3),
  list(old="d", new=4),
  list(old=NULL, new=5)
)
workplace <- list(
  input=x,
  output=rep(NA, length(x))
)

Reduce(f, subst, workplace)