hof_filter

Filter Array Column

Description

Apply an element-wise filtering function to an array column (this is essentially a dplyr wrapper for the filter(array<T>, function<T, Boolean>): array<T> built-in Spark SQL functions)

Usage

hof_filter(x, func, expr = NULL, dest_col = NULL, ...)

Arguments

Argument Description
x The Spark data frame to filter
func The filtering function
expr The array being filtered, could be any SQL expression evaluating to an array (default: the last column of the Spark data frame)
dest_col Column to store the filtered result (default: expr)
Additional params to dplyr::mutate

Examples



library(sparklyr)
sc <- spark_connect(master = "local")
# only keep odd elements in each array in `array_column`
copy_to(sc, tibble::tibble(array_column = list(1:5, 21:25))) %>%
  hof_filter(~ .x %% 2 == 1)