Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

Easily finding and replacing every match in a nested list

Take this object as an example:

expr <- substitute(mean(exp(sqrt(.)), .))

It is a nested list. I want to find every element that matches quote(.).

For example, magrittr's solution matches only the first level of the call:

dots <- c(FALSE, vapply(expr[-1], identical, quote(.), 
                        FUN.VALUE = logical(1)))
dots
[1] FALSE FALSE  TRUE

But I wanted to find every "." in an arbitrary nested list. In this particular case this would be these two dots:

expr[[3]]
expr[[2]][[2]][[2]]

And then these dots should be replaced:

expr[[3]] <- as.name("replacement")
expr[[2]][[2]][[2]] <- as.name("replacement")
expr
# mean(exp(sqrt(replacement)), replacement)

How would you do this?

like image 363
Carlos Cinelli Avatar asked Oct 02 '14 03:10

Carlos Cinelli


1 Answers

Using a recursive function:

convert.call <- function(x, replacement) {
  if (is.call(x)) as.call(lapply(x, convert.call, replacement=replacement)) else
    if (identical(x, quote(.))) as.name(replacement) else
      x
}

convert.call(expr, "x")
# mean(exp(sqrt(x)), x)
like image 129
flodel Avatar answered Oct 03 '22 10:10

flodel