r 中的数据表，将值分配给不使用其名称的特定列

Question

我正在写一个 function 如果列中的值为 NA，它将归为零。 我需要估算的表格将采用以下格式：

tab = data.table(V1 = 1, var = NA, perc = NA) 

表将具有不同的列名，但要估算的列名始终是第二个。 为简化起见，function 可以是：

impute = function(DT, variable) {
DT[is.na(get(variable)), variable := 0]
}

第二个“变量”需要包含在我认为可以工作的东西中。 我想指出

variable = colnames(tab)[2]

任何人都可以帮忙吗

Answer 1

您可以将变量包装在()中：

library(data.table)

impute = function(DT, variable) {
    DT[is.na(get(variable)), (variable) := 0]
}

variable = colnames(tab)[2]
impute(tab, variable)

tab
#   V1 var perc
#1:  1   0   NA 

Answer 2

我认为您不需要 function ，您可以

cols <- colnames(DT)[2]
DT[, (cols) := lapply(.SD, function(z) replace(z, is.na(z), 0)), .SDcols = cols]

虽然如果你想要一个，你可以做

na0 <- function(x, default = 0) replace(x, is.na(x), default)
DT[, (cols) := lapply(.SD, na0), .SDcols = cols]