是否有更优雅的过滤方式基于一组字符串中的值?
def myFilter(actions: Set[String], myDF: DataFrame): DataFrame = {
val containsAction = udf((action: String) => {
actions.contains(action)
})
myDF.filter(containsAction('action))
}
Run Code Online (Sandbox Code Playgroud)
在SQL中你可以做到
select * from myTable where action in ('action1', 'action2', 'action3')
Run Code Online (Sandbox Code Playgroud)