基于字符串是否包含所述标识符的新标识符列到数据框

Question

我是 R 的绝对新手。我想要实现的是根据同一行中的字符串值是否包含该标识符，将标识符添加到每个数据框行。

假设数据框：

df <- data.frame(Code = c("DE8230", "18FR16", "2UK34", "45BE87C", "1894DE56", "AB12FR", "ES12456"),
                 Type = c("A", "B", "C", "C", "E", "A", "C"),
                 Value = c(12, 14, 8, 20, 21, 16, 5))


      Code Type Value
1   DE8230    A    12
2   18FR16    B    14
3    2UK34    C     8
4  45BE87C    C    20
5 1894DE56    E    21
6   AB12FR    A    16
7  ES12456    C     5

我想根据标识符（例如 DE、FR、UK、BE、ES）是否存在于“代码”列中来添加一个国家/地区列，而不是列出该国家/地区。

我尝试了什么：

identifiers <- c("DE", "FR", "UK") #identifiers of choice

df <- mutate(df, country = 0)

for (i in 1:length(identifiers)){
  df <- mutate(df,
          country = ifelse(grepl(identifiers[i], Code), identifiers[i], country)
  )
}

哪个产量：

      Code Type Value country
1   DE8230    A    12      DE
2   18FR16    B    14      FR
3    2UK34    C     8      UK
4 1894DE56    C    20      DE
5   AB12FR    E    21      FR

虽然这可行，但我认为必须有一个更优雅的解决方案，省略 for 循环并只使用相同的 dplyr 语句。 但是，我一直无法弄清楚。

Nb：重要的是，提到的标识符列在单独的向量或列表中，而不是 mutate 语句的一部分。 这只是一个假设的例子，数据集和标识符的数量要大得多。

Answer 1

我们可以通过将标识符作为单个字符串与| paste来使用str_extract 分隔符并从“代码”中提取那些子字符串

library(dplyr)
library(stringr)
df %>% 
  mutate(country = str_extract(Code, str_c(identifiers, collapse = "|"))) %>% 
   drop_na(country)

-输出

      Code Type Value country
1   DE8230    A    12      DE
2   18FR16    B    14      FR
3    2UK34    C     8      UK
4 1894DE56    E    21      DE
5   AB12FR    A    16      FR