R语言循环保存模型输出报错：invalid subscript type 'S4'

数据背景 #

用户拥有如下数据框：

ourmodel difference new_topic
1        2 0.08233167         1
2        3 0.07837389         2
3        1 0.15904427         3
4        3 0.05716799         4
5        1 0.13388058         3
6        3 0.09156650         3

用户需求是按new_topic的唯一值循环拟合模型，将每个模型的precis输出结果存入out列表，但代码运行报错。

用户原代码 #

library(rethinking)

out <- list()

for (i in unique(singo$new_topic)) {
  i <- ulam(
    alist(
        difference ~ dnorm(mu, sigma),
        mu <- a[ourmodel],
        a[ourmodel] ~ dnorm(0.40, 0.15) ,
        sigma ~ dexp(2) 
    ), data = final, chains = 4, cores = 4)
  out[[i]] <- precis(i, depth = 2)
}

错误信息 #

Error in out[[i]] <- precis(i, depth = 2) : invalid subscript type 'S4'

错误原因 #

核心问题是循环变量被意外覆盖：

• 初始时i是new_topic的唯一数值（整数类型），但循环内部将i重新赋值为ulam()返回的S4类模型对象。
• 当执行out[[i]]时，试图用S4对象作为列表的下标，而R列表仅支持整数、字符或逻辑值作为下标，因此触发类型错误。

另外原代码还存在潜在问题：循环中使用data = final，但用户提供的数据源是singo，推测应该是按new_topic筛选singo的子集来拟合模型，否则每次循环都会用全量数据，失去分组拟合的意义。

修正后的代码 #

library(rethinking)

out <- list()
# 获取new_topic的唯一值集合
unique_topics <- unique(singo$new_topic)

for (topic in unique_topics) {
  # 筛选当前topic对应的数据子集
  subset_data <- singo[singo$new_topic == topic, ]
  # 拟合模型，用独立变量存储模型对象，避免覆盖循环变量
  model <- ulam(
    alist(
      difference ~ dnorm(mu, sigma),
      mu <- a[ourmodel],
      a[ourmodel] ~ dnorm(0.40, 0.15),
      sigma ~ dexp(2)
    ), 
    data = subset_data, 
    chains = 4, 
    cores = 4
  )
  # 将结果存入列表，用topic的字符形式作为命名下标，便于后续索引
  out[[as.character(topic)]] <- precis(model, depth = 2)
}

修正说明 #

1. 避免变量覆盖：将循环变量命名为topic，模型对象用model存储，彻底避免原代码中循环变量被覆盖的问题。
2. 正确筛选数据：添加subset_data步骤，确保每次循环仅使用当前new_topic对应的数据拟合模型，符合分组建模的需求。
3. 规范列表下标：用as.character(topic)作为列表的命名下标，后续可以通过out[["1"]]、out[["3"]]等方式直接索引对应topic的结果，更直观易用。

内容的提问来源于stack exchange，提问作者Quantizer