如何合并R语言DataFrame中VAR1与VAR2互为逆序的行并求和VAR3?
解决方案:合并互为逆序的行并对VAR3求和
问题分析
你需要将VAR1和VAR2互为逆序的行(如(A,B)和(B,A))合并,对VAR3求和。你之前的代码失效是因为条件逻辑错误:filter(VAR1 == VAR2)是筛选VAR1与VAR2相同的行,而你的数据中没有这类行,导致后续判断永远为假,所有VAR4都被标记为No。
简洁高效的解决方法:生成标准化分组键
核心思路是给每一对(VAR1, VAR2)生成不随顺序变化的分组键,比如按字母顺序取两个变量的最小值和最大值,这样互为逆序的行会被分到同一组,之后直接对VAR3求和即可。
代码实现
library(dplyr) # 原始数据 DF <- data.frame( VAR1 = c("A", "A", "B", "B", "B", "C", "C"), VAR2 = c("B", "C", "A", "D", "C", "B", "D"), VAR3 = c(1, 1, 1, 2, 4, 6, 4) ) # 处理步骤 result <- DF %>% # 生成标准化分组键:每行取VAR1和VAR2的最小、最大值 mutate( group_min = pmin(VAR1, VAR2), group_max = pmax(VAR1, VAR2) ) %>% # 按分组键分组,求和VAR3,并用分组键作为新的VAR1/VAR2 group_by(group_min, group_max) %>% summarise( VAR1 = group_min, VAR2 = group_max, VAR3 = sum(VAR3), .groups = "drop" ) %>% # 调整列顺序并排序(可选,匹配你的期望结果顺序) select(VAR1, VAR2, VAR3) %>% arrange(VAR1, VAR2) print(result)
运行结果
# A tibble: 5 × 3 VAR1 VAR2 VAR3 <chr> <chr> <dbl> 1 A B 2 2 A C 1 3 B C 10 4 B D 2 5 C D 4
另一种思路:自连接匹配逆序行
如果需要保留原始的VAR1/VAR2组合顺序(而不是强制按字母排序),可以用自连接的方式匹配逆序行:
result <- DF %>% # 自连接,匹配VAR1和VAR2逆序的行 left_join(DF, by = c("VAR1" = "VAR2", "VAR2" = "VAR1"), suffix = c("", "_rev")) %>% # 标记正向行(避免重复处理互为逆序的行对) mutate(is_forward = VAR1 <= VAR2) %>% # 只保留正向行或无逆序匹配的行 filter(is_forward | is.na(VAR3_rev)) %>% # 求和VAR3,无逆序则加0 mutate(VAR3 = VAR3 + coalesce(VAR3_rev, 0)) %>% # 去重并选择目标列 select(VAR1, VAR2, VAR3) %>% distinct() print(result)
内容的提问来源于stack exchange,提问作者Jeremy
相关产品推荐
相关产品推荐

