函数返回部分基于全局变量的数据集

Question

我想创建一个执行以下操作的函数：接收数据集检查全局变量x是否为0或1如果x为0则应将x转为1并返回不带第一行的数据集如果x为1然后它应该离开x并返回每一行。

数据集是

mid counts
1  -2.25      3
2  -1.75      6
3  -1.25     14
4  -0.75     14
5  -0.25     17
6   0.25     19
7   0.75     12
8   1.25      9
9   1.75      4
10  2.25      2
11  2.25     -3
12  2.75     -6
13  3.25    -14
14  3.75    -14
15  4.25    -17
16  4.75    -19
17  5.25    -12
18  5.75     -9
19  6.25     -4
20  6.75     -2

我现在的功能

testfunc = function(set){
    print(x)
    for(i in 1:nrow(set)){
           ifelse(x == 0, x <<- 1, return(set[i]))
    }
}

这样就得到了x的打印和只有数据集的计数行而没有省去第一行。

Answer 1

你的功能逻辑有点不对劲。

首先，你根本不需要for循环。

其次，您不需要ifelse来测试一个值。 length(x) == 1所以你将测试x的一个值，这应该用一个普通的if来完成。

看看这个ifelse正在做什么：

ifelse(x == 0, x <<- 1, return(set[i]))

这写着“如果x为零，则在x中将GlobalEnv设置为1，否则返回set[i]”。这显然不是你想要的。

一个正确的方法可能如下。

testfunc <- function(set){
  print(x)
  if(x == 0){
    x <<- 1
    return(set[-1, ])
  }else{
    return(set)
  }
}

现在测试它，首先使用x == 0。

x <- 0
res <- testfunc(dat)
x
#[1] 1

head(res)
#    mid counts
#2 -1.75      6
#3 -1.25     14
#4 -0.75     14
#5 -0.25     17
#6  0.25     19
#7  0.75     12

x在全局环境中的值已更改，返回的数据框没有第一行。

现在x == 1。

res <- testfunc(dat)
x
#[1] 1

head(res)
#    mid counts
#1 -2.25      3
#2 -1.75      6
#3 -1.25     14
#4 -0.75     14
#5 -0.25     17
#6  0.25     19

x的值没有改变，返回的df是原始的。