r - 使用 data.table 列中的丰富对象

Question

假设我有一个 data.table，其中一列包含线性模型:

library(data.table)
set.seed(1014)

dt <- data.table(
  g = c(1, 1, 2, 2, 3, 3, 3),
  x = runif(7),
  y = runif(7)
)

models <- dt[, list(mod = list(lm(y ~ x, data = .SD))), by = g]

现在我想从每个模型中提取 r 平方值。我可不可以做
比这个更好？

models[, list(rsq = summary(mod[[1]])$r.squared), by = g]

##    g      rsq
## 1: 1 1.000000
## 2: 2 1.000000
## 3: 3 0.004452

理想情况下，我希望能够消除 [[1]]而不是依赖
知道以前的分组变量(我知道我希望每一行都是
它是自己的组)。

Answer 1

这只是 summary作为一个糟糕的小函数，它没有被矢量化。那么如何手动对其进行矢量化(这与@mnel 的解决方案大致相同):

r.squared = Vectorize(function(x) summary(x)$r.squared)

models[, rsq := r.squared(mod)]
models
#   g  mod         rsq
#1: 1 <lm> 1.000000000
#2: 2 <lm> 1.000000000
#3: 3 <lm> 0.004451631