寻找和写作环境

如何解决寻找和写作环境

我正在编写一个与Python的“ pop”方法等效的R。我知道第99个百分位数有一个，但我更喜欢自己的百分位数（实践/理解/一致性等）。作为参考，pop（）会使用一个对象，并从对象中删除第一个项目，并返回它。所以

> l <- c(1,3,5)
> x <- pop(l)
> print(l)
> 3,5
> print(x)
> 1

我正在使用assign（）将输入对象替换为第一个值减去一个第一个值，然后从函数中返回所述第一个值。

我的问题是，如何获取输入对象的环境并在assign（）中使用此环境？我尝试使用返回{R_GlobalEnv'的pryr::where()，但是无法在assign（）中使用此值。相反，我可以在assign（）中使用的唯一值是“ globalenv（）”。

通过移动设备发布，因此如果有问题，请让我知道。

解决方法

您可以在基本R中实现此功能，尽管不建议这样做。 R是一种功能语言，最终用户不会期望具有副作用的功能。

pop <- function(vec)
{
  vec_name <- deparse(substitute(vec))
  assign(vec_name,vec[-1],envir = parent.frame())
  vec[1]
}

a <- c(2,7,9)

a
#> [1] 2 7 9

pop(a)
#> [1] 2

a
#> [1] 7 9

pop(a)
#> [1] 7

a
#> [1] 9

^{由reprex package（v0.3.0）于2020-08-15创建}

您可以使用pryr::promise_info(l)$env来做到这一点，但这是非常不像R的事情。函数不应该有副作用。

例如，

    pop <- function(l) {
      info <- pryr::promise_info(l)
      if (!is.name(info$code))
        stop("Argument expression should be a name.")
      result <- l[[1]]  # work on lists too
      assign(as.character(info$code),l[-1],envir = info$env)
      result
    }
    l <- c(1,3,5)
    pop(l)
#> Registered S3 method overwritten by 'pryr':
#>   method      from
#>   print.bytes Rcpp
#> [1] 1
    l
#> [1] 3 5

^{由reprex package（v0.3.0）于2020-08-15创建}

编辑添加：有趣的是，到目前为止，这三个答案中的任何一个都不适用于这种情况：

f <- function(x) {
  cat("The pop(x) result is",pop(x),"\n")
  cat("Now x is ",x,"\n")
  cat("Now l is ",l,"\n")
}

l <- c(1,5)
f(l)

@RuiBarradas的答案给出了

The pop(x) result is 5 
Now x is  1 3 5 
Now l is  1 3 5

（他弹出最后一个值而不是第一个值，这没什么大不了，但是x和l都没有被修改。）

@AllanCameron的答案给出了

The pop(x) result is 1 
Now x is  3 5 
Now l is  1 3 5

这可以说是正确的（x被弹出），但是我认为弹出l会很好，这似乎很棘手。

我的回答因以下消息而消失：

Error in pop(x) : Argument expression should be a name.

这似乎是个错误：显然，无论它是x还是l，它确实是一个名字。问题似乎出在pryr::promise_info中，它返回的编译代码将返回x的值，而不仅仅是返回x的代码。如果我用compiler::enableJIT(0)关闭JIT编译，则会得到与@AllanCameron相同的结果。对我来说，目前尚不清楚如何退回正确的金额以弹出l而不是x。

以下答案基于此R-Help post，函数pop和this SO post中的函数getEnvOf，两者均适用于问题的问题。

getEnvOf <- function(what,which=rev(sys.parents())) {
  what <- as.character(substitute(what))
  for (frame in which)
    if (exists(what,frame=frame,inherits=FALSE))
      return(sys.frame(frame))
  return(NULL)
}
pop <- function(x){
  y <- as.character(substitute(x))
  e <- getEnvOf(y)
  if(length(x) > 0) {
    val <- x[[length(x)]]
    assign(y,x[-length(x)],envir = parent.env(e))
    val
  } else {
    msg <- paste(sQuote(y),"length is not > 0")
    warning(msg)
    NULL
  }
}

y <- c(1,5)
pop(y)

这也适用于列表。

z <- list(1,2,5)
pop(z)

w <- list(1,c(2,4,6),5)
pop(w)
#[1] 5

pop(w)
#[1] 2 4 6

pop(w)
#[1] 1

pop(w)
#NULL
#Warning message:
#In pop(w) : ‘w’ length is not > 0

如何解决寻找和写作环境

解决方法

相关推荐