如何在R数据框中基于temp、ppt及其他列组合生成计算列W

我来帮你搞定这个生成交叉计算列的需求，这里有两种实用的实现方法，你可以根据自己的使用习惯选择：

核心思路 #

我们需要把所有以t开头的temp列和以p开头的ppt列进行两两组合，按照W = temp * ppt * area的逻辑计算新列，并给新列命名为W_<temp列名>_<ppt列名>的格式。

方法一：Base R 实现（无需额外安装包） #

先构造可重复的示例数据（加set.seed让随机结果一致，方便测试）：

set.seed(123)
df <- data.frame(id = 1:4, area = rnorm(4, 123), t_p25 = rnorm(4, 2), t_p30 = rnorm(4, 0.5), t_m25 = rnorm(4, 3), p1 = rnorm(4, 7), p_1 = rnorm(4, 10), p2 = rnorm(4, 3), p_2 = rnorm(4, -1))

然后按步骤生成所有W列：

# 筛选出所有以t开头的temp列、以p开头的ppt列
temp_cols <- grep("^t", names(df), value = TRUE)
ppt_cols <- grep("^p", names(df), value = TRUE)

# 生成temp和ppt列的所有两两组合
combinations <- expand.grid(temp = temp_cols, ppt = ppt_cols, stringsAsFactors = FALSE)

# 循环计算每个组合对应的W列
for (i in 1:nrow(combinations)) {
  temp_col <- combinations$temp[i]
  ppt_col <- combinations$ppt[i]
  new_col_name <- paste0("W_", temp_col, "_", ppt_col)
  df[[new_col_name]] <- df[[temp_col]] * df[[ppt_col]] * df$area
}

方法二：Tidyverse 实现（管道风格更简洁） #

如果你习惯用tidyverse工具链，这种方法代码更紧凑，可读性更强：

library(dplyr)
library(purrr)

set.seed(123)
df <- data.frame(id = 1:4, area = rnorm(4, 123), t_p25 = rnorm(4, 2), t_p30 = rnorm(4, 0.5), t_m25 = rnorm(4, 3), p1 = rnorm(4, 7), p_1 = rnorm(4, 10), p2 = rnorm(4, 3), p_2 = rnorm(4, -1))

# 筛选列并生成组合
temp_cols <- grep("^t", names(df), value = TRUE)
ppt_cols <- grep("^p", names(df), value = TRUE)
combinations <- expand.grid(temp = temp_cols, ppt = ppt_cols, stringsAsFactors = FALSE)

# 生成计算表达式并批量应用到数据框
calc_exprs <- map2(combinations$temp, combinations$ppt, function(temp, ppt) {
  expr(!!sym(paste0("W_", temp, "_", ppt)) := !!sym(temp) * !!sym(ppt) * area)
})

df <- df %>% mutate(!!!calc_exprs)

验证结果 #

运行完代码后，你可以用names(df)查看新增的所有W列，或者用head(df)预览数据，确认计算逻辑是否正确。

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