在函数内修改数据集但数据集未更改

Question

我的代码在 R 中如下：

replaceNA<- function(myData,limit){
    numNA<- rowsum(is.na(myData))
    targetRows<- which(numNA<=limit)
    targetCols<- length(names(myData))
    for(row in targetRows){
        for(col in 1:targetCols){
            myData[row,col][is.na(myData[row,col])]<-1
        }
    }
}

我正在尝试遍历 myData 中的每个元素，并将行的所有 NA 替换为 1，如果该行的 NA 数量不超过该行。我用 print 语句测试了我的代码，发现迭代工作完美（尽管不是最有效的代码），如果我通过在函数的最后一个括号之前放入 fix(myData) 来检查修改后的 myData，我看到我的函数完美运行（对于满足限制条件的行，NA 被替换为 1）。但是，当我在函数终止后检查 myData 时，myData 不会显示 replaceNA 所做的更改。

我知道存储修改后的 myData 存在问题，但我不确定如何正确存储它。

Answer 1

条件不清楚（英文问题）。在任何情况下，您都不需要这里的 for 循环。

计算每行缺失值的数量：

rowSums(is.na(myData))

然后您只需测试您的条件并替换所有行：

mm <- myData[rowSums(is.na(myData)) <= limit ,]
mm[is.na(mm)] <- 1
myData[rowSums(is.na(myData)) <= limit ,] <- mm

Answer 2

你应该让你的函数显式返回修改后的数据，

replaceNA<- function(myData,limit){
    numNA<- rowsum(is.na(myData))
    targetRows<- which(numNA<=limit)
    targetCols<- length(names(myData))
    for(row in targetRows){
        for(col in 1:targetCols){
            myData[row,col][is.na(myData[row,col])]<-1
        }
    }
    return(myData)
}

然后分配修改后的数据。您可以覆盖旧数据

myData <- replaceNA(myData, limit = 2)

或复制一份来比较

myData_no_na <- replaceNA(myData, limit = 2)

您也可以完全避免循环，这更像 R。 @agstudy 的回答似乎很好地涵盖了这种方法。