data.table：通过列名匹配分配给列

Question

假设我有一个数据表

> dt = data.table(Faaa = c(1,2,NA), Fbbb = c(2,NA,3), date = as.IDate(c("2010-01-01", "2010-01-02", NA)))
> dt
   Faaa Fbbb       date
1:    1    2 2010-01-01
2:    2   NA 2010-01-02
3:   NA    3       <NA>

假设有很多名称以 F 开头的列。我希望对于名称以 F 开头的每一列，用列均值来估算 NA，而不是日期（和任何其他）列。

对于这个问题，F列是相邻的，所以我可以通过索引访问，只是想知道是否有一个聪明的方法让我不需要对列索引进行硬编码。

Answer 1

您可以为此使用grep 和na.aggregate 中的na.aggregate：

library(data.table)
library(zoo)

dt <- data.table(Faaa = c(1,2,NA), Fbbb = c(2,NA,3), date = as.IDate(c("2010-01-01", "2010-01-02", NA)))

cols = grep("^F", names(dt), value = TRUE)
dt[, (cols) := lapply(.SD, na.aggregate),.SDcols = cols]

结果：

   Faaa Fbbb       date
1:  1.0  2.0 2010-01-01
2:  2.0  2.5 2010-01-02
3:  1.5  3.0       <NA>