首页 > 解决方案 > 在多个数据框中搜索包含的特定文本,并在新列中返回这些值(多次出现)

问题描述

寻找一些帮助来从一个数据框中搜索多个特定单词,在另一个数据框的列(文本正文)中,然后将这些值拉到一个新列中。

进一步解释:

以下是一些示例数据,您可以使用它们来重新创建我的问题,其中包含两个数据框:(1)test_admin我要搜索的管理员级别列表,以及 (2)test_dataset$Summary我要运行搜索的列。(您可以忽略 Other_Variables 的值,这些值在真实数据集中填充了大量值)

test_admin <- data.frame(adm1_name = c("Sindh"),
                   adm2_name = c("Central Karachi", "Dadu", "East Karachi", "Ghotki", "Sujawal", "Sukkur"))
                   
test_dataset <- data.frame(Summary = c("In Cox's Bazar, this and that happened.",
                                       "In Yangon, something else happened",
                                       "In Central Karachi, this happened",
                                       "In Sindh, this happened",
                                       "In Dadu AND East Karachi, this happened"),
                           Other_Variable_1 = 1:5,
                           Other_Variable_2 = 1:5)

为了使事情进一步复杂化,我也有兴趣从数据框的列中搜索值test_admin。例如,如果您的值“Sindh”来自 adm1_level 列,那么返回 adm2_level 下的所有结果也会非常酷。

但如果你能在更基础的层面上解决它(只搜索一列),我也会对此感到非常满意。

我要查找的输出类似于下面的数据框,它还会返回多行以显示出现多个值的位置。

                                   Summary Other_Variable_1 Other_Variable_2       Locations
1  In Cox's Bazar, this and that happened.                1                1            <NA>
2       In Yangon, something else happened                2                2            <NA>
3        In Central Karachi, this happened                3                3 Central Karachi
4                  In Sindh, this happened                4                4 Central Karachi
5                  In Sindh, this happened                4                4            Dadu
6                  In Sindh, this happened                4                4    East Karachi
7                  In Sindh, this happened                4                4          Ghotki
8                  In Sindh, this happened                4                4         Sujawal
9                  In Sindh, this happened                4                4          Sukkur
10 In Dadu AND East Karachi, this happened                5                5            Dadu
11 In Dadu AND East Karachi, this happened                5                5    East Karachi

我尝试了一些 mutate 和 grepl 函数,但没有成功。我发现的其他示例似乎仅适用于精确值或单个搜索。谢谢您的帮助!

#tidyverse 解决方案首选

标签: rsearchmatchtidyversegrepl

解决方案


这是一种方法:

library(tidyverse)

map_df(seq(nrow(test_dataset)), function(i) {
  inds <- str_detect(test_dataset$Summary[i], test_admin$adm1_name) | 
             str_detect(test_dataset$Summary[i], test_admin$adm2_name)
  if(any(inds)) tibble(test_dataset[i, ], Locations = test_admin$adm2_name[inds])
    else tibble(test_dataset[i, ], Locations = NA)
})

#  Summary                                 Other_Variable_1 Other_Variable_2 Locations      
#   <chr>                                              <int>            <int> <chr>          
# 1 In Cox's Bazar, this and that happened.                1                1 NA             
# 2 In Yangon, something else happened                     2                2 NA             
# 3 In Central Karachi, this happened                      3                3 Central Karachi
# 4 In Sindh, this happened                                4                4 Central Karachi
# 5 In Sindh, this happened                                4                4 Dadu           
# 6 In Sindh, this happened                                4                4 East Karachi   
# 7 In Sindh, this happened                                4                4 Ghotki         
# 8 In Sindh, this happened                                4                4 Sujawal        
# 9 In Sindh, this happened                                4                4 Sukkur         
#10 In Dadu AND East Karachi, this happened                5                5 Dadu           
#11 In Dadu AND East Karachi, this happened                5                5 East Karachi   

对于其中的每个值,Summary我们检查它是否匹配adm1_nameadm2_name。如果任何一行匹配,我们Location在输出中包含相应的值,否则返回NA


推荐阅读