R - 编写一个函数以使用 if 语句返回二进制输出

Question

早安，

我是一名初学者，并试图了解为什么会出现以下错误。

我正在尝试创建一个函数，该函数将根据数据集中的列值返回 0 或 1。

LT = function(Lost.time) {
    For (i in 1:dim(df)) {
        if (df$Lost.time > 0) {
            x = 1
        }
        else {
            x = 0
        }
        return(x)
    }
}

错误：没有要返回的函数，跳转到顶层另外：警告 消息：在 if (df$Lost.time > 0) { ：条件的长度 > 1 并且只有 第一个元素将被使用>}错误：“}”中的意外'}'

Answer 1

代码中有几个错误：

1. R 区分大小写。使用for 而不是For。
2. 如果您循环遍历df$Lost.time 中的条目，则应使用df$Lost.time[i] 在循环内寻址各个元素。但是，此任务不需要循环。
3. else 语句不应从代码的新行开始。解析器无法知道if 语句在第一个块之后没有完成。如果else 语句像} else { 那样用花括号括起来，那么在这个意义上不会有问题。
4. 传递给函数的参数不合适。也许您可以传递df，而不是Lost.time，但可能需要重写部分函数。
5. 在for 循环中使用1:dim(df) 应该可以工作，但会触发警告消息。最好使用1:nrow(df)。

这些是语法问题。但是，主要问题可能是@TimBiegeleisen 在回答中解决的问题：在循环中，您正在检查df$Lost.time 的每个̀nrow(df) 元素是否满足特定条件。因此，将单个二进制结果作为返回值似乎没有意义。在实现之前，应该明确功能的用途。

可以使用ifelse 在单行中构造此函数的替代方法。

Answer 2

目前尚不清楚您实际上想要在函数中返回什么。 return 只能被调用一次，之后它会返回一个值并终止函数。

如果您想获得一个包含 1 或 0 的向量，具体取决于数据框中的给定行是否具有 Lost.time > 0，那么下面的一行应该可以解决问题：

x <- as.numeric(df$Lost.time > 0)

Answer 3

如果使用循环来编写函数，则应为每个元素使用索引。 在您的数据框中创建一个变量（x），如果语句为真，则打印 1 else 0

LT = function(Lost.time) {
    for (i in 1:dim(df)) {
        if (as.numeric(df$Lost.time[i]) > 0) {
            df$x[i] <- 1
        }else{
          df$x[i] <- 0
        }
    }
}