Как назначить функцию, возвращающую более одного значения?
Попытка войти в логику 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
). Если бы это была одна и та же функция для обоих компонентов, мне не нужно было бы их разделять).
Ответы
Ответ 1
(1) list [...] & lt; - Я опубликовал это более десяти лет назад на 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
Ответ 2
Я как-то наткнулся на этот умный хак в Интернете... Я не уверен, что это противно или красиво, но это позволяет создать "волшебный" оператор, который позволяет распаковать несколько возвращаемых значений в свою собственную переменную. Функция :=
определена здесь и приведена ниже для потомков:
':=' <- 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
Я не знаю, как я к этому отношусь. Возможно, вы найдете его полезным в своем интерактивном рабочем пространстве. Использование его для сборки (перезаписываемых) библиотек (для массового потребления) может быть не лучшей идеей, но я предполагаю, что до вас.
... вы знаете, что они говорят об ответственности и власти...
Ответ 3
Обычно я переношу вывод в список, который очень гибкий (вы можете иметь любую комбинацию чисел, строк, векторов, матриц, массивов, списков, объектов, которые он выводит)
так:
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
Ответ 4
functionReturningTwoValues <- function() {
results <- list()
results$first <- 1
results$second <-2
return(results)
}
a <- functionReturningTwoValues()
Я думаю, что это работает.
Ответ 5
Я собрал 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, ..)
Проверьте пакет виньетка для получения дополнительной информации и примеров.
Ответ 6
Нет правильного ответа на этот вопрос. Я действительно зависит от того, что вы делаете с данными. В простом примере выше я настоятельно рекомендую:
- Держите все как можно проще.
- Если это возможно, рекомендуется использовать векторизацию ваших функций. Это обеспечивает максимальную гибкость и скорость в долгосрочной перспективе.
Важно ли, чтобы значения 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
[Edit] Учитывая, что вы будете применять min и max для каждого вектора отдельно, я бы предложил либо использовать матрицу (если 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
Ответ 7
Списки кажутся идеальными для этой цели. Например, внутри функции у вас будет
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]]
Ответ 8
Да на второй и третий вопросы - это то, что вам нужно сделать, поскольку вы не можете иметь несколько "lvalues" слева от задания.
Ответ 9
Как использовать назначение?
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
.
Ответ 10
[А]
Если каждый из foo и bar является единственным числом, то нет ничего плохого в c (foo, bar); и вы также можете назвать компоненты: c (Foo = foo, Bar = bar). Таким образом, вы можете получить доступ к компонентам результата "res" в качестве res [1], res [2]; или, в названном случае, в качестве res [ "Foo" ], res [ "BAR" ].
[В]
Если foo и bar являются векторами того же типа и длины, то снова нет ничего плохого в возвращении cbind (foo, bar) или rbind (foo, bar); также именный. В случае "cbind" вы получите доступ к foo и bar как res [, 1], res [, 2] или как res [, "Foo" ], res [, "Bar" ]. Возможно, вы также захотите вернуть блок данных, а не матрицу:
data.frame(Foo=foo,Bar=bar)
и получить доступ к ним как res $Foo, res $Bar. Это также хорошо работает, если foo и bar имеют одинаковую длину, но не одного типа (например, foo - вектор чисел, bar - вектор символьных строк).
[С]
Если foo и bar достаточно разные, чтобы не сочетаться удобно, как указано выше, то вы обязательно вернете список.
Например, ваша функция может соответствовать линейной модели и
также вычислить предсказанные значения, чтобы вы могли иметь
LM<-lm(....) ; foo<-summary(LM); bar<-LM$fit
а затем вы бы return list(Foo=foo,Bar=bar)
, а затем получите доступ к сводке как res $Foo, предсказанные значения как res $Bar
источник: http://r.789695.n4.nabble.com/How-to-return-multiple-values-in-a-function-td858528.html
Ответ 11
Если вы хотите вернуть результат своей функции в глобальную среду, вы можете использовать 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
Ответ 12
Чтобы получить несколько выходов из функции и сохранить их в нужном формате, вы можете сохранить выходы на свой жесткий диск (в рабочем каталоге) изнутри функции, а затем загрузить их извне функции:
myfun <- function(x) {
df1 <- ...
df2 <- ...
save(df1, file = "myfile1")
save(df2, file = "myfile2")
}
load("myfile1")
load("myfile2")
Ответ 13
С помощью R 3.6.1 я могу сделать следующее
fr2v <- function() { c(5,3) }
a_b <- fr2v()
(a_b[[1]]) # prints "5"
(a_b[[2]]) # prints "3"