R语言中为跨列无序配对的ID组合生成共享唯一CoupleID列

需求描述 #

• 数据处理流程中需新增CoupleID列：当ID与PartnerID两列的数值组合一致、顺序可互换（即两个值无需固定出现在同一列）时，为属于同一配对的所有记录分配相同的唯一ID。
• 常规分组ID生成方案仅覆盖同列内值重复的分组场景，无法满足跨列无序配对的分组需求。

原始数据示例 #

> tibble(df)
# A tibble: 6 × 2
  ID    PartnerID
   1       2        
   2       1        
   3       4        
   4       3        
   5       6        
   6       5      

期望输出效果 #

同一对双向匹配的记录共享相同CoupleID：

> tibble(df2)
# A tibble: 6 × 3
  ID    PartnerID CoupleID
   1       2         1       
   2       1         1       
   3       4         2       
   4       3         2       
   5       6         3       
   6       5         3

测试数据复现代码 #

df <- data.frame (ID  = c("1", "2", "3", "4", "5", "6"),
                  PartnerID = c("2", "1", "4","3", "6", "5")
)

df2 <- data.frame (ID  = c("1", "2", "3", "4", "5", "6"),
                  PartnerID = c("2", "1", "4","3", "6", "5"),
                  CoupleID = c("1", "1", "2", "2", "3", "3")
)

可行实现方案 #

核心逻辑通用：对每一行的ID和PartnerID值做排序，生成不受原始顺序影响的统一配对标识，再基于该标识分配连续的唯一ID即可，以下提供两种常用写法。

写法1：tidyverse 生态实现（代码可读性高） #

需要提前加载dplyr、purrr包：

library(dplyr)
library(purrr)

df <- df %>%
  mutate(
    # 逐行对两个ID排序，拼接为顺序固定的配对key
    pair_key = pmap_chr(list(ID, PartnerID), ~paste(sort(c(.x, .y)), collapse = "_")),
    # 将配对key转换为从1开始的连续整数ID
    CoupleID = as.integer(factor(pair_key))
  ) %>%
  select(-pair_key) # 移除临时生成的中间列

写法2：基础R实现（无需安装加载第三方包） #

# 逐行生成顺序固定的配对key
pair_key <- apply(df[c("ID", "PartnerID")], 1, function(row) paste(sort(row), collapse = "_"))
# 生成连续的CoupleID
df$CoupleID <- as.integer(factor(pair_key))

说明：上述代码对字符型、数值型ID均兼容；如果存在无配对的缺失值（PartnerID为NA），代码会自动为这类单条记录分配独立的CoupleID，无需额外调整逻辑。运行代码后得到的结果与期望输出完全一致。

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