将脚本更改为函数，这样它就可以循环了。脚本使用mutate和str_detect创建新的数据帧列

Question

我有一个大数据帧(462K行），其中有一个变量列出了药品的通用药品名称内容。有成千上万种不同的仿制药；我对其中大约100个感兴趣。有的药品具有多个仿制药成分；我需要知道药物中是否含有我感兴趣的药物。我创建了一个脚本，使用mutate将一个新的逻辑变量添加到数据帧中，如果使用str_detect单独或与其他仿制药组合存在一个仿制药，例如“布洛芬”，则该脚本为真。

Drug_Table_Names <- data.frame(mutate(Drug_Table_Names, DRUG_GENERIC_NAME, 
                                  Diclofenac = str_detect 
                                  (Drug_Table_Names$DRUG_GENERIC_NAME,"DICLOFEN", negate = FALSE)))

我相信我想要的函数会有两个参数：1。mutate的参数，该参数将是它将创建的变量的名称，在上面的示例中是双氯芬酸。2.str_detect的参数，用于检测，即药物的通用名称（在本例中是药物的一部分），在上面的例子中是Diclofen。

我有一个两列的tibble NSAID_LIST和两个变量，drug_flag是新变量的名称，gen_name是str_detect要查找的模式：

# A tibble: 6 x 2
  drug_flag    gen_name    
  <chr>        <chr>       
1 Diclofenac   DICLOFENAC  
2 Fenoprofen   FENOPROFEN  
3 Flurbiprofen FLURBIPROFEN
4 Ibuprofen    IBUPROFEN   
5 Ketoprofen   KETOPROFEN  
6 Naproxen     NAPROXEN    

Classes ‘tbl_df’, ‘tbl’ and 'data.frame':   13 obs. of  2 variables:
 $ drug_flag: chr  "Diclofenac" "Fenoprofen" "Flurbiprofen" "Ibuprofen" ...
 $ gen_name : chr  "DICLOFENAC" "FENOPROFEN" "FLURBIPROFEN" "IBUPROFEN" ...

下面是我编写函数的失败尝试，我称之为flagdrugnames：

FlagDrugNames<-function (drug_flag, gen_name) {Drug_Table_Names <- data.frame(mutate(Drug_Table_Names, DRUG_GENERIC_NAME, 
                                     drug_flag = str_detect 
                                     (Drug_Table_Names$DRUG_GENERIC_NAME,
                                       "gen_name", negate = FALSE)))}

FlagDrugNames(Flurbiprofen, FLURBIPROFEN)

第二：我想通过上面显示的tibble中的gen_name和drug_flag值列表运行这个函数。

如果能帮助我编写函数，然后将drug_flag和gen_name对传递给函数，我将不胜感激。

Answer 1

以mtcars集为例，可以使用以下方法：

find_names = function(dat, col1, string1, new_col_name) {

    mutate_call = lazyeval::interp(~ str_detect(a,string1), a = as.name(col1))

    dat %>% mutate_(.dots = setNames(list(mutate_call), new_col_name))

    return(dat)

  }

mtcars = find_names(mtcars, 'wt', '2', 'ibuprofen')

然后，使用for循环

for (i in 1:nrow(dict) { find_names(mtcars,'wt',dict[i,1],dict[i,2]) }