Как передать переменные по ссылке в R?

Question

Я хотел бы спросить, как я могу переписать переменные v и ind в следующей функции:

repcomb <- function(v,n,ind) 
{ 
    k <- length(v) 
    if(ind == 0) 
    { 
     for (i in 1:k) v[i] <- 1 
     ind <- 1 
     return 
    } 
    for (i in k:1) 
    { 
     if(v[i] != n) 
     { 
     for (j in k:i) v[j] <- v[i] + 1 
     return 
     } 
    } 
    ind = 0 
} 

Что это самый простой способ для обновления v и ind?

Answer 1

Это самый простой способ:

repcomb <- function(v,n,ind) 
{ 
    k <- length(v) 
    if(ind == 0) 
    { 
    for (i in 1:k) v[i] <- 1 
    ind <- 1 
    return(list(v=v, ind=ind)) 
    } 
    for (i in k:1) 
    { 
    if(v[i] != n) 
    { 
     for (j in i+1:k) v[j] <- v[i] + 1 
     v[i] <- v[i] + 1 
     return(list(v=v, ind=ind)) 
    } 
    } 
    ind = 0 
    return(list(v=v, ind=ind)) 
} 

res <- repcomb(1:5, 4, 2) 
v <- res$v 
ind <- res$ind 

Answer 2

Если вы хотите, чтобы вернуть значение параметра, установленного внутри функции, вы можете использовать eval.parent(substitute(val<-new_val)), например:

f_sqr <- function(val){ 
    new_val <- val^2 
    eval.parent(substitute(val<-new_val)) 
} 

Если вы назовите это:

val <- 5 
f_sqr(val) 
val 
#[1] 25 

Обратите внимание, что вы не должны изменять значение параметра ter внутри функции, вместо этого вы скопируете ее в новую переменную, сделайте то, что хотите сделать в своем коде, и, наконец, установите значение новой переменной в свою переменную параметра, как это происходит внутри функции.

Для собственной функции, это то, что вам нужно сделать для вашего первого if:

repcomb <- function(v, n, ind) 
{ 
    k <- length(v) 
    if (ind == 0) 
    { 
     new_v <- v 
     for (i in 1:k) new_v[i] <- 1 
     # ind <- 1 
     new_ind <- 1 
     eval.parent(substitute(ind<-new_ind)) 
     eval.parent(substitute(v<-new_v)) 
    } 
} 

Тогда, если вы звоните, вы получите изменения обратно:

v <- 1:5 
n <- 3 
ind <- 0 
repcomb(v, n, ind) 
v 
#[1] 1 1 1 1 1 
ind 
#[1] 1 

Соответственно, другая часть может быть изменена в соответствии с тем, что вы хотите.