Question

我正在尝试编写一个函数，该函数将在数据帧中搜索给定的字符串。例如：

# Set up
library(tidyverse)
options(stringsAsFactors= F)

color <- c('black', 'black', 'blue', 'blue', 'yellow')
vehicle <- c('truck', 'truck', 'car', 'car', 'bike')
plant <- c('tree', 'flower', 'grass', 'tree', 'flower')
height <- c('tall', 'medium', 'short', 'tall', 'medium')

testdf <- as.data.frame(cbind(color, vehicle, plant, height))

使函数可以在任何行中搜索具有“卡车”值的任何变量：

search.func <- function(df) {
  names(df %>%
    select_if(is.character) %>%
    select_if(grepl('truck', .)))
}

search.func(testdf)  # returns the correct result - 'vehicle'

为了使功能更灵活，并能够传递任何字符串，我尝试过：

search.func2 <- function(df, string) {

  string <- enquo(string)

  names(df %>%
          select_if(is.character) %>%
          select_if(grepl(string, .)))
  }

search.func2(testdf, truck)  # errors out

但是我没有正确使用enquo-我需要grepl函数中的引号，并且我很难告诉R如何做到这一点。任何帮助表示赞赏！谢谢！

Answer 1

添加quo_name之后，您就可以了。

search.func2 <- function(df, string) {

  string <- enquo(string)
  string <- quo_name(string)
  names(df %>%
          select_if(is.character) %>%
          select_if(grepl(string, .)))
}

#search.func2(testdf, truck)
#[1] "vehicle"

自定义函数中的数据框字符串搜索

1 个答案: