在 R 中本地（而非全局）设置种子

Question

我只想在 R 中本地设置种子（在函数内部），但似乎 R 不仅在本地设置种子，而且在全局设置种子。这是我正在尝试（不）做的一个简单示例。

myfunction <- function () {
  set.seed(2)
}

# now, whenever I run the two commands below I'll get the same answer
myfunction()
runif(1)

所以，我的问题是：为什么 R 在全局设置种子，而不仅仅是在我的函数内部？我怎样才能让 R 仅在我的函数内设置种子？

Answer 1

这样的事情对我来说是这样的：

myfunction <- function () {
  old <- .Random.seed
  set.seed(2)
  res <- runif(1)
  .Random.seed <<- old
  res
}

或者也许更优雅：

myfunction <- function () {
  old <- .Random.seed
  on.exit( { .Random.seed <<- old } )
  set.seed(2)
  runif(1)
}

例如：

> myfunction()
[1] 0.1848823
> runif(1)
[1] 0.3472722
> myfunction()
[1] 0.1848823
> runif(1)
[1] 0.4887732

Answer 2

使用@Romain Francois的答案，概括为函数：

withRandom <- function(expr, seed = 1) {
    old <- .Random.seed
    on.exit({.Random.seed <<- old})
    set.seed(seed)
    expr
}

用途：

runif(2)
withRandom(seed = 2, {
    runif(1)
    runif(1)
})
runif(2)
withRandom(seed = 2, runif(2))
runif(2)

输出：

> runif(2)
[1] 0.5776099 0.6309793
> withRandom(seed = 2, {
+     runif(1)
+     runif(1)
+ })
[1] 0.702374
> runif(2)
[1] 0.5120159 0.5050239
> withRandom(seed = 2, runif(2))
[1] 0.1848823 0.7023740
> runif(2)
[1] 0.5340354 0.5572494

Answer 3

要在本地为一段代码设置种子，您可以使用

with_seed

包中的

withr

函数：

withr::with_seed(123, {
  runif(1) # this always gives the same number
})
runif(1)   # this one is different every time