R data.table - 如何在多列的固定行上插入缺失值

Question

a = data.table(c(2,NA,3), c(5,NA,1))

当我尝试对缺失的行进行插值时

a[, approx(x = 1:.N, y = .SD, xout = which(is.na(.SD))), .SDcols = 1:2]

出现以下错误:

Error in xy.coords(x, y) : 'x' and 'y' lengths differ

我希望得到以下信息:

> a
    V1 V2
1: 2.0  5
2: 2.5  3
3: 3.0  1

Answer 1

x 和 y(前两个参数)似乎应该是数值向量。您需要遍历每一列。这里我使用 set() 和 for-loop 来更新原始 data.table 通过引用.

len = 1:nrow(a)
for (col in names(a)) {
    nas = which(is.na(a[[col]]))
    set(a, i=nas, j=col, value=approx(len, a[[col]], xout=nas)$y)
}
#     V1 V2
# 1: 2.0  5
# 2: 2.5  3
# 3: 3.0  1