R语言中用For循环自动化计算Dataframe均值的函数问题

问题背景 #

以下计算DataFrame行均值的代码可以正常运行：

res <- list()
for(i in 1:nrow(df))
  z <- df[i,]
  z <- unlist(z) 
  z <- mean(z)
  res[[i]] <- z
  print(res)

但改写为函数后，输出列表中除最后一个元素外其余均为NULL：

res <- list()
col.mean <- function(x){
  for(i in 1:nrow(x))
    z <- x[i,]
    z <- unlist(z) 
    z <- mean(z)
    res[[i]] <- z
    return(res)
}

下方简化后的函数可正常运行，但逻辑看似相近：

res <- list()
col.mean <- function(x){
  for(i in 1:nrow(x))
    res[[i]] <- mean(unlist(x[i,]))
    return(res)
}

一、第一个函数异常的原因 #

核心问题有两个：

• for循环体未用大括号包裹：R语言中，如果for语句后没有用{}，只会把紧跟在for行的那一行代码当作循环体执行。第一个函数里，循环只会重复执行z <- x[i,]这一行，把x的每一行依次赋值给z，但后面的unlist、mean、res[[i]] <- z这些代码，只会在循环跑完所有i之后，以最后一次循环的i值（即nrow(x)）执行一次，所以只有res的最后一个位置被赋值，其余位置都是初始的NULL。
• 依赖全局变量+隐性执行顺序错误：函数直接修改全局定义的res，且后续赋值、返回代码仅执行一次，前面的循环迭代完全没起到赋值作用。

二、简化后函数的运行逻辑 #

这个函数能正常工作的关键在于：

1. 循环体为单一行代码：res[[i]] <- mean(unlist(x[i,]))是紧跟for的唯一一行，所以循环会遍历i从1到nrow(x)的所有值，依次计算第i行的均值并赋值给res的第i个元素。
2. return语句在循环外：return(res)在循环的下一级缩进，会等整个循环执行完毕后才返回结果，因此res的所有位置都能被正确赋值。

更规范的写法建议 #

为了避免歧义，最好用大括号明确包裹循环体，同时将res的初始化放在函数内部，不依赖全局变量：

col.mean <- function(x){
  res <- list()
  for(i in 1:nrow(x)){
    res[[i]] <- mean(unlist(x[i,]))
  }
  return(res)
}

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