嵌套循环中创建函数遇问题：无法将结果写入向量

你的函数存在的核心问题 #

1. 函数未接收参数，依赖全局变量：first()直接使用全局环境的i、j、data，耦合性高，不符合函数封装的设计逻辑，也容易引发变量作用域问题。
2. 未接收函数返回值：调用first()时没有将返回值赋值给test，导致test始终是空向量。
3. 函数内的赋值+return逻辑冗余：return(test <- c(test, ...))试图在函数内修改全局test，但R函数默认在局部环境运行，这种写法既不规范也容易出错。

修正后的代码方案 #

第一步：封装独立的比较函数

让函数只负责接收两个死亡值，返回比较结果，不依赖任何全局变量，逻辑更清晰：

compare_death <- function(death_i, death_j) {
  if (death_i < death_j) {
    return(-1)
  } else if (death_i > death_j) {
    return(1)
  } else {
    return(0)
  }
}

第二步：重构循环逻辑

提前提取干预组的死亡数据，减少重复索引查询，循环中调用函数并拼接结果：

data <- data.frame(
  id = c(1, 5, 6, 11, 15, 21),
  intervention = c(2, 2, 2, 1, 1, 1),
  death = c(0, 1, 0, 1, 0, 0)
)

# 提前提取两组的id和对应death值
group1_ids <- data$id[data$intervention == 1]
group1_deaths <- data$death[data$intervention == 1]
group2_ids <- data$id[data$intervention == 2]
group2_deaths <- data$death[data$intervention == 2]

for (k in seq_along(group1_ids)) {
  current_id <- group1_ids[k]
  current_death <- group1_deaths[k]
  cat(paste0("id = ", current_id, ": "), "\n")
  
  test <- c()
  for (l in seq_along(group2_ids)) {
    # 调用比较函数，将结果添加到test
    result <- compare_death(current_death, group2_deaths[l])
    test <- c(test, result)
  }
  
  print(test)
}

额外优化建议 #

1. 避免循环内拼接向量：R中test <- c(test, result)每次都会重新分配内存，数据量大时效率低，建议预先初始化向量：

# 提前初始化指定长度的空向量
test <- integer(length(group2_ids))
for (l in seq_along(group2_ids)) {
  test[l] <- compare_death(current_death, group2_deaths[l])
}

2. 用outer()替代嵌套循环：代码更简洁，效率更高：

# 直接生成所有组1和组2的比较结果矩阵
comparison_matrix <- outer(group1_deaths, group2_deaths, compare_death)

# 遍历矩阵每行打印对应id的结果
for (k in seq_along(group1_ids)) {
  cat(paste0("id = ", group1_ids[k], ": "), "\n")
  print(comparison_matrix[k, ])
}

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