在 R 函数中使用“prob”包时出错

Question

我正在尝试编写一个使用prob 包来计算条件概率的函数。使用该函数时，我继续遇到相同的错误，即找不到函数内的对象。

下面是一个可重现的示例，其中我在没有函数的情况下计算条件概率，然后尝试使用该函数产生相同的结果。我不确定错误是由于 prob 包的限制还是我的错误。

# Load prob package
library(prob)

# Set seed for reproducibility
set.seed(30)

# Sample data frame
sampledata <- data.frame(
  X <- sample(1:10),
  Y <- sample(c(-1, 0, 1), 10, replace=TRUE))

# Set probability space
S <- probspace(sampledata)

# Subset Y between -1 and 0
A <- subset(S, Y>=-1 & Y<=0)

# Subset X greater than 6
B <- subset(S, X>6)

# Compute conditional probability
P <- prob(A, given=B)

以上代码产生以下概率：

> P
[1] 0.25

尝试编写一个函数来计算相同的概率：

# Create function with data frame, variables, and conditional inputs
prob.function <- function(df, variable1, variable2, state1, state2, cond1){
  s <- probspace(df)
  a <- subset(s, variable1>=state1 & variable1<=state2)
  b <- subset(s, variable2>cond1)
  p <- prob(a, given=b)
  return(p)
}

# Demonstrate the function
test <- prob.function(sampledata, Y, X, -1, 0, 6)

此函数给出以下错误：

Error in eval(expr, envir, enclos) : object 'b' not found

您能提供的任何帮助都会很棒。

谢谢！

Answer 1

这看起来像是prob 中的一个错误。

当我在 Vanilla R 中运行它时，我得到了同样的错误。但是当我在我的工作区创建一个对象b 时，错误消失了：

> print(b)
Error in print(b) : object 'b' not found
> test <- prob.function(sampledata, Y, X, -1, 0, 6)
Error in eval(expr, envir, enclos) : object 'b' not found
>
> b <- "dummy variable"
> print(b)
[1] "dummy variable"
> test <- prob.function(sampledata, Y, X, -1, 0, 6)
> test
[1] 0.25
>

作为一种临时解决方法，只需在当前环境中创建一个虚拟 b。

---

至于错误，如果您查看prob.default 的源代码（在上面的示例中，prob(a, given=b) 最终会调用它），您将看到以下部分：

if (missing(given)) {
    < cropped >
}
else {
    f <- substitute(given)
    g <- eval(f, x)                  <~~~~ 
    if (!is.logical(g)) {            <~~~~
        if (!is.data.frame(given))   <~~~~
            stop("'given' must be data.frame or evaluate to logical")
        B <- given

    }
    ...
    < cropped >
}

它从g 跳到given，也许是无意的？我会联系包维护者，因为这可能是一个疏忽。

Answer 2

我不认为这是包 prob 中的错误。

首先，您应该将sampledata 创建为

sampledata <- data.frame(
  X = sample(1:10),
  Y = sample(c(-1, 0, 1), 10, replace=TRUE))

您的原始代码不仅会创建此数据帧，还会在全局环境中创建变量 X 和 Y，这些变量会在您稍后调用函数时实际使用。

其次，你不应该在函数内部调用subset()。改用括号子集：

prob.function <- function(df, variable1, variable2, state1, state2, cond1){
  s <- probspace(df)
  a <- s[s[[variable1]]>=state1 & s[[variable1]]<=state2, ]
  b <- s[s[[variable2]]>cond1, ]
  p <- prob(a, given=b)
  return(p)
}

并将variable1 和variable2 作为字符串传递：

test <- prob.function(sampledata, "Y", "X", -1, 0, 6)

现在你有test==0.25，没有错误。

参考资料：

• http://adv-r.had.co.nz/Computing-on-the-language.html#non-standard-evaluation-in-subset
• Assignment operators in R: '=' and '<-'
• Why is `[` better than `subset`?