为 data.table 的列设置多个不同的属性

Question

data.table 具有优雅的setattr 用于将单个属性就地添加到列。有没有一种优雅的方法可以一步覆盖多个属性？例如，假设data.table 有很多列，我想为列x1 分配两个属性，为列x3 分配三个属性，这可能在以下list 中指定：

a <- list(x1=list(label='X1', units='mm'),
          x3=list(label='X3', comment='collected remotely', format='type 3'))

我可以轻松编写处理a 并调用setattr 5 次的代码来完成此操作。但我希望有更好的方法。

Answer 1

我不知道下面的代码是否非常优雅，但它可以工作。这是一个双 *apply 循环。
引用问题：

我可以轻松编写处理a 并调用setattr 5 次的代码来完成此操作。但我希望有更好的方法。

问题是setattr 中的name 必须是长度为1 的字符串，所以setattr 总是要被调用5 次。在下面的代码中，这是伪装成双循环完成的。

例子data.table来自help("setattr")中的第三个DT。

library(data.table)

DT <- data.table(x1 = 1:3, y = 4:6, x3 = 7:9)
a <- list(x1=list(label='X1', units='mm'),
          x3=list(label='X3', comment='collected remotely', format='type 3'))

mapply(function(x, a){
  lapply(names(a), function(na) setattr(DT[[x]], na, a[[na]]))
}, names(a), a)

attributes(DT$x1)
#$label
#[1] "X1"
#
#$units
#[1] "mm"

attributes(DT$x3)
#$label
#[1] "X3"
#
#$comment
#[1] "collected remotely"
#
#$format
#[1] "type 3"

---

注意。为了避免循环输出丑陋，请将它们包装在invisible中：

invisible(
  mapply(function(x, a){
    lapply(names(a), function(na) setattr(DT[[x]], na, a[[na]]))
  }, names(a), a)
)

---

编辑

下面的代码比较简单。

lapply(names(a), function(x){
  lapply(names(a[[x]]), function(y) setattr(DT[[x]], y, a[[x]][[y]]))
})

Answer 2

这可能与您想要的输出偏差太大，但只是抛出一个想法：因为setattr 接受data.table，另一种方法可能是在data.table 级别设置属性，作为指向的命名列表到各个列：

setattr(d, "all_attr", a)
str(d)
# Classes ‘data.table’ and 'data.frame':    3 obs. of  3 variables:
# $ x1: int  1 2 3
# $ y : int  4 5 6
# $ x3: int  7 8 9
# - attr(*, ".internal.selfref")=<externalptr> 
#   - attr(*, "all_attr")=List of 2
# ..$ x1:List of 2
# .. ..$ label: chr "X1"
# .. ..$ units: chr "mm"
# ..$ x3:List of 3
# .. ..$ label  : chr "X3"
# .. ..$ comment: chr "collected remotely"
# .. ..$ format : chr "type 3"

---

如果您希望在单个列级别设置属性，并且您可以将属性作为嵌套列表使用，我认为循环遍历列可能就足够了。

lapply(names(a), function(x) setattr(d[[x]], x, a[[x]]))
str(d)
# Classes ‘data.table’ and 'data.frame':    3 obs. of  3 variables:
# $ x1: int  1 2 3
# ..- attr(*, "x1")=List of 2
# .. ..$ label: chr "X1"
# .. ..$ units: chr "mm"
# $ y : int  4 5 6
# $ x3: int  7 8 9
# ..- attr(*, "x3")=List of 3
# .. ..$ label  : chr "X3"
# .. ..$ comment: chr "collected remotely"
# .. ..$ format : chr "type 3"
# - attr(*, ".internal.selfref")=<externalptr>

---

library(data.table)
d = data.table(x1 = 1:3, y = 4:6, x3 = 7:9)