如何使用tidyverse风格实现tibble全列的字符串拼接泛化操作

实现方案 #

你可以用dplyr的行操作语法或者purrr的逐行映射功能实现需求，符合tidyverse代码风格，无需手动写循环：

1. 可读性优先版本（基于rowwise）

library(tidyverse)

generate_concat_text <- function(df) {
  # 拆分前n-1列和最后一列
  pre_cols <- head(names(df), -1)
  last_col <- tail(names(df), 1)
  
  df %>%
    rowwise() %>%
    transmute(
      text = str_c(
        # 前n-1列按「列名_列值」格式拼接，用空格分隔
        str_c(pre_cols, "_", c_across(all_of(pre_cols)), collapse = " "),
        # 最后一列前加换行符直接拼接
        "\n",
        .data[[last_col]]
      )
    ) %>%
    ungroup()
}

2. 性能优先版本（基于purrr::pmap）

适合处理大数据量的场景，不需要逐行分组：

generate_concat_text_purrr <- function(df) {
  pre_cols <- head(names(df), -1)
  last_col <- tail(names(df), 1)
  
  df %>%
    mutate(
      text = pmap_chr(
        .l = pick(everything()),
        .f = function(...) {
          row_vals <- list(...)
          pre_part <- str_c(pre_cols, "_", row_vals[pre_cols], collapse = " ")
          str_c(pre_part, "\n", row_vals[[last_col]])
        }
      )
    ) %>%
    select(text)
}

验证测试 #

和你给出的示例代码输出完全一致：

# 构造测试数据
starwars_with_species_as_last_column <- starwars %>% 
  select(1:11)

# 调用函数
function_result <- generate_concat_text(starwars_with_species_as_last_column)

# 原代码输出
original_result <- starwars_with_species_as_last_column %>% 
  transmute(text = str_c("gender_", gender, " homeworld_", homeworld, "\n", species))

# 一致性校验
identical(function_result, original_result)
# 运行返回 TRUE

说明 #

• 这里用.data[[列名]]实现动态列引用，是tidyverse推荐的动态变量操作方案，不需要手动写准引用逻辑
• 如果要自定义前n-1列的分隔符、最后一列的前缀，只需要修改str_c的参数即可

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