Как назначить из функции, которая возвращает более одного значения? - PullRequest
198 голосов
/ 01 декабря 2009

Все еще пытаясь разобраться в логике R ... каков "лучший" способ распаковать (на LHS) результаты функции, возвращающей несколько значений?

Я не могу этого сделать, по-видимому:

R> functionReturningTwoValues <- function() { return(c(1, 2)) }
R> functionReturningTwoValues()
[1] 1 2
R> a, b <- functionReturningTwoValues()
Error: unexpected ',' in "a,"
R> c(a, b) <- functionReturningTwoValues()
Error in c(a, b) <- functionReturningTwoValues() : object 'a' not found

Должен ли я действительно делать следующее?

R> r <- functionReturningTwoValues()
R> a <- r[1]; b <- r[2]

или программист на R напишет что-то вроде этого:

R> functionReturningTwoValues <- function() {return(list(first=1, second=2))}
R> r <- functionReturningTwoValues()
R> r$first
[1] 1
R> r$second
[1] 2

--- отредактировано, чтобы отвечать на вопросы Шейна ---

Мне не нужно давать имена частям значения результата. Я применяю одну агрегатную функцию к первому компоненту, а другую - ко второму компоненту (min и max. Если бы это была одна и та же функция для обоих компонентов, мне не нужно было бы их разбивать).

Ответы [ 12 ]

168 голосов
/ 28 февраля 2013

(1) list [...] <- </strong> Я опубликовал это более десяти лет назад на r-help . С тех пор он был добавлен в пакет gsubfn. Он не требует специального оператора, но требует, чтобы левая часть была написана с использованием list[...], например:

library(gsubfn)  # need 0.7-0 or later
list[a, b] <- functionReturningTwoValues()

Если вам нужен только первый или второй компонент, все они тоже работают:

list[a] <- functionReturningTwoValues()
list[a, ] <- functionReturningTwoValues()
list[, b] <- functionReturningTwoValues()

(Конечно, если вам нужно только одно значение, то functionReturningTwoValues()[[1]] или functionReturningTwoValues()[[2]] будет достаточно.)

См. Цитируемую ветку r-help для получения дополнительных примеров.

(2) с Если цель состоит в том, чтобы просто объединить несколько значений последовательно, а возвращаемые значения были названы, тогда простой альтернативой является использование with:

myfun <- function() list(a = 1, b = 2)

list[a, b] <- myfun()
a + b

# same
with(myfun(), a + b)

(3) attach Другой альтернативой является attach:

attach(myfun())
a + b

ДОБАВЛЕНО: with и attach

68 голосов
/ 02 декабря 2009

Я как-то наткнулся на этот хитрый взлом в интернете ... Я не уверен, насколько он неприятен или красив, но он позволяет вам создать "магический" оператор, который позволяет вам распаковать несколько возвращаемых значений в их собственную переменную. := функция определена здесь и включена ниже для потомков:

':=' <- function(lhs, rhs) {
  frame <- parent.frame()
  lhs <- as.list(substitute(lhs))
  if (length(lhs) > 1)
    lhs <- lhs[-1]
  if (length(lhs) == 1) {
    do.call(`=`, list(lhs[[1]], rhs), envir=frame)
    return(invisible(NULL)) 
  }
  if (is.function(rhs) || is(rhs, 'formula'))
    rhs <- list(rhs)
  if (length(lhs) > length(rhs))
    rhs <- c(rhs, rep(list(NULL), length(lhs) - length(rhs)))
  for (i in 1:length(lhs))
    do.call(`=`, list(lhs[[i]], rhs[[i]]), envir=frame)
  return(invisible(NULL)) 
}

Имея это в руках, вы можете делать то, что вам нужно:

functionReturningTwoValues <- function() {
  return(list(1, matrix(0, 2, 2)))
}
c(a, b) := functionReturningTwoValues()
a
#[1] 1
b
#     [,1] [,2]
# [1,]    0    0
# [2,]    0    0

Я не знаю, что я чувствую по этому поводу. Возможно, вы найдете это полезным в вашем интерактивном рабочем пространстве. Использование его для создания (повторного) использования библиотек (для массового потребления) может быть не самой лучшей идеей, но я думаю, это ваше дело.

... вы знаете, что они говорят об ответственности и власти ...

42 голосов
/ 01 декабря 2009

Обычно я заключаю вывод в список, который очень гибок (вы можете иметь любую комбинацию чисел, строк, векторов, матриц, массивов, списков, объектов в выводе)

так как:

func2<-function(input) {
   a<-input+1
   b<-input+2
   output<-list(a,b)
   return(output)
}

output<-func2(5)

for (i in output) {
   print(i)
}

[1] 6
[1] 7
14 голосов
/ 23 марта 2016
functionReturningTwoValues <- function() { 
  results <- list()
  results$first <- 1
  results$second <-2
  return(results) 
}
a <- functionReturningTwoValues()

Я думаю, что это работает.

11 голосов
/ 17 июля 2017

Я собрал пакет R фанат , чтобы решить эту проблему. Zeallot включает в себя оператор множественного присваивания или распаковки присваивания, %<-%. LHS оператора - это любое количество переменных для назначения, построенных с использованием вызовов c(). RHS оператора - это вектор, список, фрейм данных, объект даты или любой пользовательский объект с реализованным методом destructure (см. ?zeallot::destructure).

Вот несколько примеров, основанных на оригинальном сообщении,

library(zeallot)

functionReturningTwoValues <- function() { 
  return(c(1, 2)) 
}

c(a, b) %<-% functionReturningTwoValues()
a  # 1
b  # 2

functionReturningListOfValues <- function() {
  return(list(1, 2, 3))
}

c(d, e, f) %<-% functionReturningListOfValues()
d  # 1
e  # 2
f  # 3

functionReturningNestedList <- function() {
  return(list(1, list(2, 3)))
}

c(f, c(g, h)) %<-% functionReturningNestedList()
f  # 1
g  # 2
h  # 3

functionReturningTooManyValues <- function() {
  return(as.list(1:20))
}

c(i, j, ...rest) %<-% functionReturningTooManyValues()
i     # 1
j     # 2
rest  # list(3, 4, 5, ..)

Проверьте пакет виньетка для получения дополнительной информации и примеров.

10 голосов
/ 01 декабря 2009

Нет правильного ответа на этот вопрос. Я действительно зависит от того, что вы делаете с данными. В приведенном выше простом примере я настоятельно рекомендую:

  1. Делайте вещи максимально простыми.
  2. По возможности, рекомендуется сохранять векторизацию ваших функций. Это обеспечивает наибольшую гибкость и скорость в долгосрочной перспективе.

Важно ли, чтобы значения 1 и 2 выше имели имена? Другими словами, почему в этом примере важно, чтобы 1 и 2 были названы a и b, а не просто r [1] и r [2]? В этом контексте важно понимать, что a и b равны и обоим векторам длины 1. Таким образом, вы ничего не меняете в процессе назначения, кроме двух новых векторов, которые не нужно ссылаться на подписки:

> r <- c(1,2)
> a <- r[1]
> b <- r[2]
> class(r)
[1] "numeric"
> class(a)
[1] "numeric"
> a
[1] 1
> a[1]
[1] 1

Вы также можете присвоить имена исходному вектору, если вы предпочитаете указывать букву, а не индекс:

> names(r) <- c("a","b")
> names(r)
[1] "a" "b"
> r["a"]
a 
1 

[Редактировать] Учитывая, что вы будете применять мин и макс к каждому вектору в отдельности, я бы предложил либо использовать матрицу (если a и b будут одинаковой длины и одинакового типа данных), либо фрейм данных (если a и b будут одинаковой длины, но могут иметь разные типы данных) или использовать список, как в вашем последнем примере (если они могут иметь различную длину и типы данных).

> r <- data.frame(a=1:4, b=5:8)
> r
  a b
1 1 5
2 2 6
3 3 7
4 4 8
> min(r$a)
[1] 1
> max(r$b)
[1] 8
5 голосов
/ 28 февраля 2013

Списки кажутся идеальными для этой цели. Например, внутри функции у вас будет

x = desired_return_value_1 # (vector, matrix, etc)

y = desired_return_value_2 # (vector, matrix, etc)

returnlist = list(x,y...)

}  # end of function

основная программа

x = returnlist[[1]]

y = returnlist[[2]]
3 голосов
/ 01 декабря 2009

Да, на ваш второй и третий вопросы - это то, что вам нужно сделать, поскольку у вас не может быть нескольких «lvalues» слева от задания.

2 голосов
/ 04 июля 2014

Как насчет использования assign?

functionReturningTwoValues <- function(a, b) {
  assign(a, 1, pos=1)
  assign(b, 2, pos=1)
}

Вы можете передавать имена переменных, которые вы хотите передать по ссылке.

> functionReturningTwoValues('a', 'b')
> a
[1] 1
> b
[1] 2

Если вам нужен доступ к существующим значениям, обратное значение assign равно get.

1 голос
/ 30 апреля 2017

Если вы хотите вернуть вывод вашей функции в Глобальную среду, вы можете использовать list2env, как в этом примере:

myfun <- function(x) { a <- 1:x
                       b <- 5:x
                       df <- data.frame(a=a, b=b)

                       newList <- list("my_obj1" = a, "my_obj2" = b, "myDF"=df)
                       list2env(newList ,.GlobalEnv)
                       }
    myfun(3)

Эта функция создаст три объекта в вашей глобальной среде:

> my_obj1
  [1] 1 2 3

> my_obj2
  [1] 5 4 3

> myDF
    a b
  1 1 5
  2 2 4
  3 3 3
...