Как связать аргументы функции - PullRequest
4 голосов
/ 01 июля 2011

Как мне частично связать / применить аргументы к функции в R?

Это то, как далеко я продвинулся, тогда я понял, что этот подход не работает ...

bind <- function(fun,...)
{
  argNames <- names(formals(fun))
  bindedArgs <- list(...)
  bindedNames <- names(bindedArgs)
  function(argNames[!argNames %in% bindedArgs])
   {
   #TODO
  }
}

Спасибо!

Ответы [ 3 ]

8 голосов
/ 04 июля 2011

Вот версия Curry, которая сохраняет ленивую оценку аргумента функции, но создает функцию, которая печатает в меру красиво:

Curry <- function(FUN, ...) {
  args <- match.call(expand.dots = FALSE)$...
  args$... <- as.name("...")

  env <- new.env(parent = parent.frame())

  if (is.name(FUN)) {
    fname <- FUN
  } else if (is.character(FUN)) {
    fname <- as.name(FUN)
  } else if (is.function(FUN)){
    fname <- as.name("FUN")
    env$FUN <- FUN
  } else {
    stop("FUN not function or name of function")
  }
  curry_call <- as.call(c(list(fname), args))

  f <- eval(call("function", as.pairlist(alist(... = )), curry_call))
  environment(f) <- env
  f
}

Это в основном работает, генерируя анонимную функцию точно так же, как если бы вы сами создавали частичное связывание.

5 голосов
/ 01 июля 2011

На самом деле, это похоже на работу вокруг

bind <- function(fun,...)
{
  boundArgs <- list(...)
  formals(fun)[names(boundArgs)] <- boundArgs
  fun
}

Однако в идеале я хочу, чтобы связанные аргументы полностью исчезли из новой функции, чтобы вызовы новой функции могли происходить со спецификацией имени, например, с add <- function(a,b) a+b я бы хотел (bind(add,a=2))(1) вернуть 3.

4 голосов
/ 01 июля 2011

Вы пробовали смотреть на функцию карри roxygen?

> library(roxygen) 
> Curry
function (FUN, ...) 
{
    .orig = list(...)
    function(...) do.call(FUN, c(.orig, list(...)))
}
<environment: namespace:roxygen>

Пример использования:

> aplusb <- function(a,b) {
+   a + 2*b
+ }
> oneplusb <- Curry(aplusb,1)
> oneplusb(2)
[1] 5

Edit: Curry кратко определен, чтобы принимать именованные или безымянные аргументы, но частичное применение fun к аргументам посредством присваивания formal() требует более сложного сопоставления для эмуляции той же функциональности. Например:

> bind <- function(fun,...)
+ {
+   argNames <- names(formals(fun))
+   boundArgs <- list(...)
+   boundNames <- names(boundArgs)
+   if(is.null(boundNames)) {
+     formals(fun)[1:length(boundArgs)] <- boundArgs
+   } else {
+     formals(fun)[match(names(boundArgs),argNames)] <- boundArgs
+   }
+   fun
+ }
> oneplusb <- bind(aplusb,1)
> oneplusb(2)
Error in 2 * b : 'b' is missing

Поскольку первый аргумент в этой функции по-прежнему a, необходимо указать, для какого аргумента 2 предназначен (b=), или передать его в качестве второго аргумента.

> oneplusb
function (a = 1, b) 
{
    a + 2 * b
}
> oneplusb(b=2) ## or oneplusb(,2)
[1] 5
...