R语言:如何判断数据框列是否包含向量中的任意元素
解决方法:判断逗号分隔字符串是否包含目标元素
我明白你的问题了——你遇到的核心问题是df$Gamma里的每一行都是单个逗号分隔的字符串,而不是可直接用%in%判断的元素向量,而且原代码里的any()会把整个列作为一个整体判断,导致所有行得到相同的错误结果。下面给你几种可行的解决方案:
首先先还原你的数据,方便测试:
df <- data.frame( Alpha = 1:3, Beta = c(100, 120, 150), Gamma = c("Dollar, Euro, Yen", "Euro", "Yen, Peso"), stringsAsFactors = FALSE ) currencies <- c("Dollar", "Yen")
方法1:用正则表达式快速匹配(最简便)
利用stringr包的str_detect或者base R的grepl,通过正则表达式匹配任意一个目标货币:
# 用stringr包 library(stringr) df$Epsilon <- as.integer(str_detect(df$Gamma, paste(currencies, collapse = "|"))) # 或者base R版本 df$Epsilon <- as.integer(grepl(paste(currencies, collapse = "|"), df$Gamma))
原理:paste(currencies, collapse = "|")会生成正则表达式Dollar|Yen,表示匹配"任意一个出现的目标货币";str_detect/grepl检查每行字符串是否包含该模式,返回TRUE/FALSE,最后用as.integer把布尔值转为1/0。
方法2:拆分字符串后严谨判断(避免部分匹配)
如果担心出现部分匹配的情况(比如字符串里有"Yenny"这种不想被误判的内容),可以先把每行的字符串拆分成元素向量,再检查是否和目标货币有交集:
# 用purrr + stringr的版本 library(purrr) library(stringr) df$Epsilon <- as.integer(map_lgl(df$Gamma, function(x) { # 拆分字符串并去除每个元素的空格 curr_list <- str_split(x, ", ")[[1]] # 检查是否有元素在目标列表中 any(curr_list %in% currencies) })) # 或者base R版本 df$Epsilon <- sapply(df$Gamma, function(x) { curr_list <- strsplit(x, ", ")[[1]] as.integer(any(curr_list %in% currencies)) })
两种方法最终都会得到正确的Epsilon列:
| Alpha | Beta | Gamma | Epsilon |
|---|---|---|---|
| 1 | 100 | Dollar, Euro, Yen | 1 |
| 2 | 120 | Euro | 0 |
| 3 | 150 | Yen, Peso | 1 |
内容的提问来源于stack exchange,提问作者SeeDerekEngineer
相关产品推荐
相关产品推荐

