r语言 - 部分基于全局变量返回数据集的函数



我想做一个函数来执行以下操作: 获取数据集 检查全局变量 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 的打印,并且只打印数据集的计数行,而没有遗漏第一行。

你的函数的逻辑有点错误。

首先,您根本不需要for循环。

其次,您不需要ifelse来测试一个值。length(x) == 1,因此您将测试x的一个值,这应该使用普通if来完成。

看看这个ifelse在做什么:

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

其内容为"如果x为零,则在GlobalEnv否则返回set[i]"中将x设置为 1"。这显然不是你想要的。

正确的方法如下。

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 是原始值。

最新更新