You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多

如何将dplyr::reframe生成的新数据框与原DataFrame重新合并?

解决方法

要保留原数据列并关联扩展后的体重级别检查结果,有三种常用的tidyverse实现方式:

方法一:在reframe中显式保留原列

因为rowwise()会将每一行视为独立分组,直接在reframe里指定需要保留的原数据列(如name、mass),这些列会自动重复对应到每个扩展行,无需后续合并:

boxingPopularWeightClasses <- c(57,80,92,9999) #kilograms
boxingWeightClassName<-c("Featherweight","Middleweight","Heavyweight","Super Heavyweight")

checkAllowedToCompeteInEachWeightClass <- function(mass) {
  mass < boxingPopularWeightClasses
}

library(dplyr)

allowed <- starwars %>% 
  rowwise() |>
  reframe(
    # 保留需要的原数据列
    name = name,
    mass = mass,
    class = boxingWeightClassName,
    criterion = boxingPopularWeightClasses, 
    allowed = checkAllowedToCompeteInEachWeightClass(mass)
  ) 

head(allowed)

输出会直接包含原数据的name、mass列,以及每个角色的4行体重级别检查结果。

方法二:用mutate生成列表列再unnest(更灵活)

无需rowwise(),先通过mutate为每行生成包含所有体重级别结果的列表列,再用unnest展开,原数据的所有列都会完整保留:

library(dplyr)
library(tidyr)

allowed <- starwars %>%
  mutate(
    # 为每个mass生成嵌套的体重检查结果表
    weight_check = pmap(
      list(mass),
      function(m) {
        tibble(
          class = boxingWeightClassName,
          criterion = boxingPopularWeightClasses,
          allowed = m < boxingPopularWeightClasses
        )
      }
    )
  ) %>%
  # 展开嵌套列表,生成多行结果
  unnest(weight_check)

head(allowed)

这种方式会保留starwars原有的所有列(如height、hair_color等),同时每个角色对应4行体重级别检查记录。

方法三:通过唯一ID事后合并(适合已生成分离结果的场景)

如果已经用reframe生成了不含原列的结果,可以给原数据添加唯一标识,再通过标识关联:

# 给原数据添加唯一行ID
starwars_with_id <- starwars %>% mutate(row_id = row_number())

# 生成结果时带上ID
allowed <- starwars_with_id %>% 
  rowwise() |>
  reframe(
    row_id = row_id,
    class = boxingWeightClassName,
    criterion = boxingPopularWeightClasses, 
    allowed = checkAllowedToCompeteInEachWeightClass(mass)
  )

# 合并原数据与结果
final_result <- starwars_with_id %>% left_join(allowed, by = "row_id")

内容的提问来源于stack exchange,提问作者Alex Holcombe

相关产品推荐
方舟 Agent Plan

超全模态模型 × Harness 升级,最新支持 Deepseek-V4.1-Flash、GLM-5.3 系列、Doubao-Seedream-5.0-pro、Kimi-K3 (部分), 限时 9.9 元起

最近更新时间:2026.06.19 07:45:10