如何根据值列表过滤RDD

Question

我能够使用特定值

完成RDD [Test]的过滤器

Test.filter(_.state == "NY") 

我需要对此进行扩展，以便能够使用多个值（“NY”，“CA”，“PA”）过滤测试

scala中是否有“in list”或（“NY”，“CA”，“PA”）

Answer 1

您可以将值设置为Set并按以下方式过滤：

val set = Set("NY", "CA", "PA")
rdd.filter(item => set(item.state))

测试值是否已设置：

set.contains(value) 

或

set(value)