根据模式匹配添加列

Question

我有一个包含两列的数据框，并且想根据第一列的值填充第二列（当前为空）。 如果第一列中的任何位置都有“ _ab”，“ _ cd”或“ _ef”，则“ ab”，“ cd”或“ ef”分别应在第二列中。

这就是它的外观：

c1         c2
s_d_ab     ab
a_cd_aa    cd
s_sar_ef   ef

在Excel中，我会复制公式=IF(FIND("_ab",A1), "ab", "") ，然后隐藏不匹配的行，并用=IF(FIND("_ab",A1), "ab", "") 。 等等。 不是最优雅的方法，但是它可以工作。

在R中解决此问题的最佳方法是什么？

我设法获得了适用条件的逻辑值向量（ grepl("_abc", data$c1) ），但不知道如何设置第二列的值。

Answer 1

您可以使用sub函数：

x <- c('s_d_ab', 'a_cd_aa', 's_sar_ef')
sub('.*_(ab|cd|ef).*', '\\1', x)
# [1] "ab" "cd" "ef"