Условия суммы в выражении R

Question

Учитывая выражение R, которое представляет собой сумму таких терминов, как

expr <- expression(a + b * c + d + e * f)

Я хотел бы получить набор всех условий суммы в виде списка имен или выражений. Таким образом, в примере элементы будут: a, b * c, d и e * f.

Из комментария следует следующее.

Сами команды могут содержать оператор +, как в

expression(a + b * c + d + e * (f + g))

, поэтому нам нужно некоторое понимание языка R.

Есть ли простой способ, например, использовать call_tree из пакета pryr ?

Answer 1

Вы можете использовать рекурсивную функцию для сканирования дерева разбора:

foo <- function(x) {
  if (is.call(x)) y <- as.list(x) else return(x)

  #stop crawling if not a call to `+`
  if (y[[1]] != quote(`+`)) return(x) 

  y <- lapply(y, foo)

  return(y[-1]) #omit `+` symbol
}

expr1 <- expression(a + b * c + d + e * f)
unlist(foo(expr1[[1]]))
#[[1]]
#a
#
#[[2]]
#b * c
#
#[[3]]
#d
#
#[[4]]
#e * f


expr2 <- expression(a + b * c + d + e * (f + g))
unlist(foo(expr2[[1]]))
#[[1]]
#a
#
#[[2]]
#b * c
#
#[[3]]
#d
#
#[[4]]
#e * (f + g)

Answer 2

Попробуйте рекурсивно разобрать выражение:

getTerms <- function(e, x = list()) {
  if (is.symbol(e)) c(e, x)
  else if (identical(e[[1]], as.name("+"))) Recall(e[[2]], c(e[[3]], x))
  else c(e, x)
}

expr <- expression(a + b * c + d + e * (f + g))
getTerms(expr[[1]])

давая:

[[1]]
a

[[2]]
b * c

[[3]]
d

[[4]]
e * (f + g)