基于 R 中的输入参数的 mtext() 的选择性文本？

Question

我有一个矢量化的 R 函数，可以生成一些直方图（如下）。我想在其中一个直方图中添加一个mtext()（请在下面找到它）。

问题

但我希望mtext() 中的文本根据参数n 是长度> 1 的向量还是参数es 是长度> 1 的向量来更改。

在我的代码中，我使用以下代码没有成功：

txt = if(length(n) > 1) "Group Sample Size = " else if(length(es) > 1) "Effect Size = " else "Group Sample Size = "

mtext(paste0(txt, input), 3) [请在下面找到这些]

这是我完整的 R 代码：

 p.es = function(n, es, n.sim){

input = if(length(n) > 1) n else if(length(es) > 1) es else n

t.sim = Vectorize(function(n, es){

d = numeric(n.sim)
p = numeric(n.sim)

for(i in 1:n.sim){
   N = sqrt((n^2)/(2*n))
   x = rnorm(n, es, 1)
   y = rnorm(n, 0, 1)
   a = t.test(x, y, var.equal = TRUE)
d[i] = a[[1]]/N
p[i] = a[[3]] }

txt = if(length(n) > 1) "Group Sample Size = " else if(length(es) > 1) "Effect Size = " else "Group Sample Size = "
hist(p) ; mtext(paste0(txt, input), 3) ## Here I need help!!!!

hist(d) 
}, c("n", "es"))

par(mfcol = c(2, length(input)), xpd = NA)
invisible(t.sim(n, es)) }
# Example of use:
p.es(n = 20, es = c(.1, .2), n.sim = 20) # Here I expect the mtext to show "Effect Size = "
                                         # but it doesn't why?

Answer 1

我认为问题不在于mtext 的if...else 语句。看起来问题是因为Vectorize 而发生的。让我们从您的内部函数开始，这里称为myfun。它似乎工作得很好。

myfun <- function(n, es, nsim, input){

  d = numeric(n.sim)
  p = numeric(n.sim)

  for(i in 1:n.sim){
    N = sqrt((n^2)/(2*n))
    x = rnorm(n, es, 1)
    y = rnorm(n, 0, 1)
    a = t.test(x, y, var.equal = TRUE)
    d[i] = a[[1]]/N
    p[i] = a[[3]] }

  txt <- if (length(n) > 1) {"Group Sample Size = "} else if (length(es) > 1) {"Effect Size = "} else {"Group Sample Size = "}
  hist(p) ; mtext(paste0(txt, input, collapse = " "), 3) ## Here I need help!!!!

  hist(d) 
}

# Test Run
par(mfrow = c(1,2))
myfun(n = 10, 
      es = c(0.1, 0.2), 
      nsim = 10, 
      input = c(0.1, 0.2))

现在，如果你实现Vectorize，就会出现问题。

t.sim <- Vectorize(myfun, c('n', 'es', 'nsim', 'input'))
n <- 10
n.sim <- 10
es <- c(0.1, 0.2)
input <- c(0.1, 0.2)
t.sim(n, es, n.sim, input)

解决此问题的一种方法是将txt 分配移到Vectorize 调用之外（您可以在定义input 时分配txt 的值吗？）。

p.es = function(n, es, n.sim){

  input = if(length(n) > 1) n else if(length(es) > 1) es else n
  txt = if(length(n) > 1) "Group Sample Size = " else if(length(es) > 1) "Effect Size = " else "Group Sample Size = "

  t.sim = Vectorize(function(n, es){
    d = numeric(n.sim)
    p = numeric(n.sim)

    for(i in 1:n.sim){
      N = sqrt((n^2)/(2*n))
      x = rnorm(n, es, 1)
      y = rnorm(n, 0, 1)
      a = t.test(x, y, var.equal = TRUE)
      d[i] = a[[1]]/N
      p[i] = a[[3]] }

    hist(p) ; mtext(paste0(txt, paste(input, collapse = "-")), 3) ## Here I need help!!!!

    hist(d) 
  }, c("n", "es"))

  par(mfcol = c(2, length(input)), xpd = NA)
  invisible(t.sim(n, es)) }
# Example of use:
p.es(n = 20, es = c(.1, .2), n.sim = 20) # Is this what you want?