R语言基于两列匹配从另一个数据框取值填充ifelse中的NA值
解决方案
直接补全现有代码空缺
你现有代码中ifelse的空缺位置可以填入匹配计算逻辑,完整可运行代码如下:
library(dplyr) a <- a %>% mutate(age_diff = ifelse( is.na(age_diff), b$base_age[match(paste(id, group), paste(b$id, b$group))] - age, age_diff ))
上述代码通过match匹配id+group的组合,取出b表对应的base_age后减去当前行的age,得到需要填充的差值。
更推荐的dplyr标准写法
如果不限制写法,推荐先做左连接再用coalesce填充空值,逻辑更清晰可读性更强:
library(dplyr) a <- a %>% # 按id、group两个字段关联两个表 left_join(b, by = c("id", "group")) %>% # 优先保留原age_diff的非空值,空值用base_age减age的结果填充 mutate(age_diff = coalesce(age_diff, base_age - age)) %>% # 移除关联引入的base_age字段,恢复a表原有结构 select(-base_age)
两种写法运行后得到的结果都和你给出的期望输出完全一致。
内容的提问来源于stack exchange,提问作者jackDanielle
相关产品推荐
相关产品推荐

