如何在R循环中获取遍历的数据框列名？

R和Python遍历数据框的行为确实不一样——Python直接遍历df会返回列名，但R默认遍历的是数据框的列值（每个列对应的向量），这就是你之前代码没得到预期结果的原因。下面给你两种在循环内获取列名的可行方案：

方案1：直接遍历列名向量 #

最直接的方式是遍历names(df)或者colnames(df)（两者在数据框上效果一致），这样循环变量就是列名，还能通过列名轻松获取对应列的数据：

for (col_name in names(df)) {
  # 打印列名
  print(col_name)
  # 后续复杂操作示例：计算该列的均值
  col_mean <- mean(df[[col_name]], na.rm = TRUE)
  print(paste("列", col_name, "的均值为：", col_mean))
}

方案2：通过索引遍历（同时获取列名和列数据） #

如果你习惯用索引的方式遍历，也可以用seq_along(df)生成列的索引，再通过索引获取列名：

for (i in seq_along(df)) {
  # 通过索引获取列名
  col_name <- names(df)[i]
  print(col_name)
  # 对应的列数据可以用 df[[i]] 或者 df[[col_name]] 获取
  col_sum <- sum(df[[i]], na.rm = TRUE)
  print(paste("列", col_name, "的总和为：", col_sum))
}

为什么你之前的代码无效？ #

你写的for (i in df)里，i是数据框每一列的向量（比如df$列名对应的向量），而name(i)不是R的正确函数（应该是names()），就算用names(i)，也只会返回该向量的元素名（如果有的话），而不是数据框的列名，所以无法得到你想要的结果。

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