问题描述
我正在编写一个与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