Filter data by levels of categorical variables

This function filters a dataframe or survey design object by keeping only the rows where a specified categorical variable matches one of the given levels. The resulting filtered dataframe is returned, along with the tidyverse code used to generate it.

Usage

filter_cat(data, var, levels)

Arguments

data

A dataframe or survey design object to be filtered.

var

The name of the column in data to be filtered by.

levels

A character vector of levels in var to keep.

Value

A filtered dataframe with the tidyverse code attached.

See also

code

Author

Owen Jin, Zhaoming Su

Examples

filtered <- filter_cat(iris,
    var = "Species",
    levels = c("versicolor", "virginica")
)
cat(code(filtered))
#> iris |> dplyr::filter(Species %in% c("versicolor", "virginica"))
head(filtered)
#>   Sepal.Length Sepal.Width Petal.Length Petal.Width    Species
#> 1          7.0         3.2          4.7         1.4 versicolor
#> 2          6.4         3.2          4.5         1.5 versicolor
#> 3          6.9         3.1          4.9         1.5 versicolor
#> 4          5.5         2.3          4.0         1.3 versicolor
#> 5          6.5         2.8          4.6         1.5 versicolor
#> 6          5.7         2.8          4.5         1.3 versicolor